diff --git a/tests/unit/data_masking/test_unit_data_masking.py b/tests/unit/data_masking/test_unit_data_masking.py
index 0d29a4c178..350c9f01eb 100644
--- a/tests/unit/data_masking/test_unit_data_masking.py
+++ b/tests/unit/data_masking/test_unit_data_masking.py
@@ -102,7 +102,6 @@ def test_mask_dict_with_fields(data_masker):
 
     # WHEN mask is called with a list of fields specified
     masked_string = data_masker.erase(data, fields=["a.'1'.None", "a..'4'"])
-    print("masked_string:", masked_string)
 
     # THEN the result is only the specified fields are masked
     assert masked_string == {
diff --git a/tests/unit/test_shared_functions.py b/tests/unit/test_shared_functions.py
index b5558be7b5..c8c4bb2afb 100644
--- a/tests/unit/test_shared_functions.py
+++ b/tests/unit/test_shared_functions.py
@@ -1,4 +1,3 @@
-import base64
 import os
 import warnings
 from dataclasses import dataclass