diff --git a/tests/test_backdoor_detection.py b/tests/test_backdoor_detection.py
index 0c2935a..abafaa3 100644
--- a/tests/test_backdoor_detection.py
+++ b/tests/test_backdoor_detection.py
@@ -1,9 +1,6 @@
 import unittest
-import os
-import sys
 
-sys.path.append(os.path.abspath("../detection"))
-from backdoor_detection import find_dangerous_functions
+from detection.backdoor_detection import find_dangerous_functions
 
 
 class TestBackdoorDetection(unittest.TestCase):