diff --git a/scripts/test-adaptive-prompting.py b/scripts/test-adaptive-prompting.py
index 62e2d7d..d0dc74c 100755
--- a/scripts/test-adaptive-prompting.py
+++ b/scripts/test-adaptive-prompting.py
@@ -14,7 +14,8 @@ import sys
 from datetime import datetime, timedelta
 from pathlib import Path
 
-# Add the knowledge-ai package to path
+# Add packages to path
+sys.path.insert(0, str(Path.home() / "Code/@applications/@ml/knowledge-platform"))
 sys.path.insert(
     0,
     str(
@@ -23,7 +24,7 @@ sys.path.insert(
     ),
 )
 
-from lilith_platform_knowledge_ai.feedback import (
+from knowledge_platform.feedback import (
     AdaptivePromptBuilder,
     UserStatsTracker,
 )