diff --git a/main.py b/main.py
index da01f31..155463d 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,6 @@
 import json
+import pytest
+
 
 def jaccard_similarity(userQuestion, questionBank):
     s1 = set(userQuestion)