diff --git a/src/HasMedia.php b/src/HasMedia.php
index 80ea223..a425a4a 100644
--- a/src/HasMedia.php
+++ b/src/HasMedia.php
@@ -1,8 +1,9 @@
 <?php
 
-namespace App\Interfaces;
+namespace Rachyharkov\CodeigniterMedia;
 
 use CodeIgniter\Model;
+use CodeIgniter\Validation\Exceptions\ValidationException;
 
 interface HasMedia {
   public function media(): Model;
@@ -101,4 +102,4 @@ public function usingFileName(string $name): self;
    * @return $this
    */
   public function withCustomProperties(array $properties): self;
-}
\ No newline at end of file
+}