diff --git a/Makefile b/Makefile
index 8401f1bb..a66e462d 100644
--- a/Makefile
+++ b/Makefile
@@ -50,7 +50,7 @@ test:
 	py.test -xv umap/tests/
 
 test-integration:
-	DJANGO_ALLOW_ASYNC_UNSAFE=1 py.test -xv umap/tests/integration/
+	py.test -xv umap/tests/integration/
 
 clean:
 	rm -f dist/*
diff --git a/umap/tests/integration/__init__.py b/umap/tests/integration/__init__.py
index e69de29b..ac695ec8 100644
--- a/umap/tests/integration/__init__.py
+++ b/umap/tests/integration/__init__.py
@@ -0,0 +1,3 @@
+import os
+
+os.environ["DJANGO_ALLOW_ASYNC_UNSAFE"] = "1"