diff --git a/conftest.py b/conftest.py
index 85e5e6b85928a22d3233e56e0b2d90e08b188700..4da087be50a7ae29695276e93811de79aff77b5f 100644
--- a/conftest.py
+++ b/conftest.py
@@ -33,6 +33,8 @@ add_path_to_ignore('pystencils_tests/benchmark')
 add_path_to_ignore('_local_tmp')
 
 
+collect_ignore += [os.path.join(SCRIPT_FOLDER, "pystencils/autodiff.py")]
+
 try:
     import pycuda
 except ImportError:
diff --git a/pystencils/autodiff.py b/pystencils/autodiff.py
index 5e185835ae03fb8883f6c932f7fb2e7a17a73463..96ac836c82542fb66dad74401233a86afd9a1bf7 100644
--- a/pystencils/autodiff.py
+++ b/pystencils/autodiff.py
@@ -8,7 +8,5 @@ Installation:
 .. code-block:: bash
     pip install pystencils-autodiff
 """
-import os
 
-if 'CI' not in os.environ:
-    raise NotImplementedError('pystencils-autodiff is not installed. Run `pip install pystencils-autodiff`')
+raise NotImplementedError('pystencils-autodiff is not installed. Run `pip install pystencils-autodiff`')