diff --git a/setupext.py b/setupext.py
index 953ec0d16405..1e27211e2415 100644
--- a/setupext.py
+++ b/setupext.py
@@ -495,12 +495,12 @@ def get_package_data(self):
                 'tests/test_*.ipynb',
             ],
             'mpl_toolkits': [
-                *_pkg_data_helper('mpl_toolkits/axes_grid1',
-                                  'tests/baseline_images'),
-                *_pkg_data_helper('mpl_toolkits/axisartist'
-                                  'tests/baseline_images'),
-                *_pkg_data_helper('mpl_toolkits/mplot3d'
-                                  'tests/baseline_images'),
+                *_pkg_data_helper('mpl_toolkits',
+                                  'axes_grid1/tests/baseline_images'),
+                *_pkg_data_helper('mpl_toolkits',
+                                  'axisartist/tests/baseline_images'),
+                *_pkg_data_helper('mpl_toolkits',
+                                  'mplot3d/tests/baseline_images'),
             ]
         }