diff --git a/tests/prompt.nim b/tests/prompt.nim
index 07e73939..76ac60aa 100644
--- a/tests/prompt.nim
+++ b/tests/prompt.nim
@@ -1,6 +1,7 @@
 import utils/utils
-import ../src/[db, directorypath, prompt, resultcode]
+import ../src/db
 import unittest2
+include ../src/prompt
 
 suite "Unit tests for prompt module":