diff --git a/tests/test_backdoor_detection.py b/tests/test_backdoor_detection.py
index c0d2a05..3d4fcf5 100644
--- a/tests/test_backdoor_detection.py
+++ b/tests/test_backdoor_detection.py
@@ -1,7 +1,7 @@
 import unittest
 import warnings
 
-from detection.backdoor_detection import find_dangerous_functions
+from detection.__main__ import find_dangerous_functions
 from detection.GPTdetection import detectGPT
 import os
 
@@ -83,5 +83,6 @@ class TestBackdoorDetection(unittest.TestCase):
         self.assertEqual(len(results["medium"]), 0)
         self.assertEqual(len(results["low"]), 0)
 
+
 if __name__ == "__main__":
     unittest.main()