diff --git a/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py index 3f7c2252f..a51b6354d 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py +++ b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py @@ -1,4 +1,4 @@ -from . import SizeError +from .errors import SizeError class EncryptionKey32Bytes(bytes): diff --git a/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py b/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py index 12aab72ba..241292e0d 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py +++ b/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py @@ -3,7 +3,7 @@ import logging from cryptography.fernet import Fernet -from . import EncryptionKey32Bytes +from .encryption_key_types import EncryptionKey32Bytes from .i_encryptor import IEncryptor logger = logging.getLogger(__name__)