diff --git a/demo/CoLA/requirements.txt b/demo/CoLA/requirements.txt
new file mode 100644
index 0000000..df907c2
--- /dev/null
+++ b/demo/CoLA/requirements.txt
@@ -0,0 +1 @@
+keras-bert
diff --git a/demo/SST-2/requirements.txt b/demo/SST-2/requirements.txt
new file mode 100644
index 0000000..df907c2
--- /dev/null
+++ b/demo/SST-2/requirements.txt
@@ -0,0 +1 @@
+keras-bert
diff --git a/test.sh b/test.sh
index 21dd8a9..2187ef6 100755
--- a/test.sh
+++ b/test.sh
@@ -1,3 +1,3 @@
 #!/usr/bin/env bash
 pycodestyle --max-line-length=120 keras_xlnet tests demo && \
-    nosetests --nocapture --with-coverage --cover-erase --cover-html --cover-html-dir=htmlcov --cover-package=keras_xlnet --with-doctest
\ No newline at end of file
+    nosetests --nocapture --with-coverage --cover-erase --cover-html --cover-html-dir=htmlcov --cover-package=keras_xlnet --with-doctest tests
\ No newline at end of file