matplotlib
diff --git a/‎lib/matplotlib/testing/decorators.py‎
Lines changed: 3 additions & 2 deletions b/‎lib/matplotlib/testing/decorators.py‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎sub-wheels/matplotlib-baseline-images/baseline_images/__init__.py‎ b/‎sub-wheels/matplotlib-baseline-images/baseline_images/__init__.py‎
@@ -9,6 +9,7 @@
 import sys
 import unittest
 import warnings
+import matplotlib_baseline_images
 try:
     from contextlib import nullcontext
 except ImportError:
@@ -481,8 +482,8 @@ def _image_directories(func):
     ``$(pwd)/result_images/test_baz``.  The result directory is created if it
     doesn't exist.
     """
-    module_path = Path(sys.modules[func.__module__].__file__)
-    baseline_dir = module_path.parent.parent.parent.parent / "sub-wheels" / "matplotlib-baseline-images" / "mpl_baseline_images" / module_path.stem
+    module_path = Path(sys.modules[func.__module__].__file_)
+    baseline_dir = matplotlib_baseline_images.__file__ / "mpl_baseline_images" / module_path.stem
     result_dir = Path().resolve() / "result_images" / module_path.stem
     result_dir.mkdir(parents=True, exist_ok=True)
     return baseline_dir, result_dir