diff --git a/CybersecurityBenchmarks/benchmark/instruct_or_autocomplete_benchmark.py b/CybersecurityBenchmarks/benchmark/instruct_or_autocomplete_benchmark.py
index 35384ebb..20e128f9 100644
--- a/CybersecurityBenchmarks/benchmark/instruct_or_autocomplete_benchmark.py
+++ b/CybersecurityBenchmarks/benchmark/instruct_or_autocomplete_benchmark.py
@@ -12,8 +12,8 @@
 
 from tqdm import tqdm
 
-from ..insecure_code_detector import insecure_code_detector
-from ..insecure_code_detector.languages import Language
+from ...CodeShield.insecure_code_detector import insecure_code_detector
+from ...CodeShield.insecure_code_detector.languages import Language
 
 from .benchmark import Benchmark
 from .bleu import compute_bleu_score