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 b8da397b5..424b66f3d 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
@@ -30,17 +30,15 @@ class KeyBasedEncryptor(IEncryptor):
     def encrypt(self, plaintext: bytes) -> bytes:
         """
         Encrypts a given bytestream
-        :param bytes plaintext: The bytestream to encrypt
+        :param plaintext: The bytestream to encrypt
         :return: Encrypted message
-        :rtype: bytes
         """
         return self._fernet.encrypt(plaintext)
 
     def decrypt(self, ciphertext: bytes) -> bytes:
         """
         Decrypts a given bytestream
-        :param bytes ciphertext: The bytestream to decrypt
+        :param ciphertext: The bytestream to decrypt
         :return: Decrypted message
-        :rtype: bytes
         """
         return self._fernet.decrypt(ciphertext)