1 files changed, 5 insertions, 2 deletions
diff --git a/test/test_init.py b/test/test_init.py
index b717e35..0915f13 100644
--- a/test/test_init.py
+++ b/test/test_init.py
@@ -1,9 +1,12 @@
-from obfuscator.__init__ import main
+from gdpr_obfuscator import Obfuscator
+import moto
+
+obfuscator = Obfuscator()
 
 
 def test_main_integration():
     test_source = "s3://test-bucket/data.csv"
-    result = main(test_source, ["name"], log_level="DEBUG")
+    result = obfuscator.process_s3(test_source, ["name"])
     result_str = result.decode("utf-8")
     assert "John" not in result_str
     assert "***" in result_str