diff --git a/image_cipher/tests/test_cipher.py b/image_cipher/tests/test_cipher.py
index d503466..3b3164b 100644
--- a/image_cipher/tests/test_cipher.py
+++ b/image_cipher/tests/test_cipher.py
@@ -1,7 +1,7 @@
 import pytest
 import os
 from cryptography.fernet import Fernet
-from image_cipher import ImageCipher
+from image_cipher.cipher import ImageCipher
 
 @pytest.fixture
 def image_path():
@@ -70,4 +70,4 @@ def test_decode_with_incorrect_key(self, image_path, encryption_key):
         encoded_path = cipher.encode(image_path, message, encrypt=True)
         incorrect_key = Fernet.generate_key()
         with pytest.raises(ValueError, match="The provided key is incorrect."):
-            cipher.decode(encoded_path, incorrect_key)
\ No newline at end of file
+            cipher.decode(encoded_path, incorrect_key)