diff --git a/script/bootstrap_server b/script/bootstrap_server
index c68e198c0149d864ce88da433f4074b9adaa082d..8d71e01fa78d29fca56346cc639294173e7d3d17 100755
--- a/script/bootstrap_server
+++ b/script/bootstrap_server
@@ -7,4 +7,4 @@ echo "Installing dependencies..."
 python3 -m pip install --upgrade -r requirements_all.txt
 
 echo  "Installing development dependencies.."
-python3 -m pip install --upgrade flake8 pylint coveralls pytest
+python3 -m pip install --upgrade flake8 pylint coveralls pytest pytest-cov
diff --git a/script/test b/script/test
index 753ec340fd6b669b951c0464527cd95e3ce3b733..56fe4dcec898b7ecf1bc7b6469855bb586679532 100755
--- a/script/test
+++ b/script/test
@@ -5,12 +5,12 @@
 
 cd "$(dirname "$0")/.."
 
+script/lint
+
 echo "Running tests..."
 
 if [ "$1" = "coverage" ]; then
-  coverage run -m unittest discover tests
+  py.test --cov homeassistant tests
 else
-  python3 -m unittest discover tests
+  py.test tests
 fi
-
-script/lint