diff --git a/Makefile b/Makefile
index e077f4b..ece9db0 100644
--- a/Makefile
+++ b/Makefile
@@ -4,15 +4,11 @@ COVERAGE_MODE := atomic
 
 test:
 	@echo "mode: $(COVERAGE_MODE)" > $(COVERAGE_REPORT); \
-	go test . -coverprofile=$(COVERAGE_PROFILE) -covermode=${COVERAGE_MODE); \
-	if [ $$? != 0 ]; then \
-		exit 2; \
-	fi; \
 	if [ -f $(COVERAGE_PROFILE) ]; then \
 		tail -n +2 $(COVERAGE_PROFILE) >> $(COVERAGE_REPORT); \
 		rm $(COVERAGE_PROFILE); \
 	fi; \
-	for dir in `find ./generator -name "*.go" | grep -o '.*/' | sort -u`; do \
+	for dir in `find . -name "*.go" | grep -o '.*/' | sort -u`; do \
 		go test $$dir -coverprofile=$(COVERAGE_PROFILE) -covermode=$(COVERAGE_MODE); \
 		if [ $$? != 0 ]; then \
 			exit 2; \