diff --git a/Makefile b/Makefile
index aae9d77bb0..c434fc19c9 100644
--- a/Makefile
+++ b/Makefile
@@ -4,6 +4,8 @@
 export SLURM_LOCALID=0
 # assume you have installed need packages
 export SPHINX_MOCK_REQUIREMENTS=1
+# install only Lightning Trainer packages
+export PACKAGE_NAME=pytorch
 
 clean:
 	# clean all temp runs
@@ -31,7 +33,7 @@ test: clean
 	pip install -e . -r requirements/pytorch/devel.txt
 	pip install -r requirements/pytorch/strategies.txt
 	# run tests with coverage
-	python -m coverage run --source pytorch_lightning -m pytest pytorch_lightning tests -v
+	python -m coverage run --source src/pytorch_lightning -m pytest src/pytorch_lightning tests/tests_pytorch -v
 	python -m coverage report
 
 docs: clean