diff --git a/script/bootstrap_server b/script/bootstrap_server
index c68e198c014..8d71e01fa78 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 753ec340fd6..56fe4dcec89 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