fixed

samadpls · Jul 14, 2024 · 2009e2b · 2009e2b
1 parent 9e4b75e
commit 2009e2b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)
+            cipher.decode(encoded_path, incorrect_key)