diff --git a/test/test_generate_claim_and_question.py b/test/test_generate_claim_and_question.py
index 7ccdd47..4714dc8 100644
--- a/test/test_generate_claim_and_question.py
+++ b/test/test_generate_claim_and_question.py
@@ -4,7 +4,7 @@
 
 import unittest
 import os
-from generate_claim import extract_atomic_claims, generate_questions
+from data.generate_claim import extract_atomic_claims, generate_questions
 current_path = os.path.dirname(os.path.abspath(__file__))
 main_folder_path = os.path.join(current_path, '..')