diff --git a/arviz/tests/test_data.py b/arviz/tests/test_data.py
index 4725aac82e..ad140ab06b 100644
--- a/arviz/tests/test_data.py
+++ b/arviz/tests/test_data.py
@@ -17,6 +17,7 @@
     load_cached_models,
     BaseArvizTest,
     pystan_extract_unpermuted,
+    pystan_extract_normal,
 )