diff --git a/recipe.yaml b/recipe.yaml
index 1a0fc6a53a..4e82d4996d 100644
--- a/recipe.yaml
+++ b/recipe.yaml
@@ -5,6 +5,7 @@ context:
   # Read version from _version.txt file generated by setuptools_scm
   version: ${{ load_from_file("_version.txt") | default("0.0.0.dev0") }}
   python_min: "3.10"
+  module_name: simpeg
 
 package:
   name: ${{ name|lower }}
@@ -47,31 +48,35 @@ requirements:
 tests:
   - python:
       imports:
-        - simpeg
-        - simpeg.electromagnetics
-        - simpeg.electromagnetics.analytics
-        - simpeg.electromagnetics.frequency_domain
-        - simpeg.electromagnetics.natural_source
-        - simpeg.electromagnetics.natural_source.utils
-        - simpeg.electromagnetics.static
-        - simpeg.electromagnetics.static.resistivity
-        - simpeg.electromagnetics.static.induced_polarization
-        - simpeg.electromagnetics.static.spectral_induced_polarization
-        - simpeg.electromagnetics.static.utils
-        - simpeg.electromagnetics.time_domain
-        - simpeg.electromagnetics.utils
-        - simpeg.electromagnetics.viscous_remanent_magnetization
-        - simpeg.flow
-        - simpeg.flow.richards
-        - simpeg.potential_fields
-        - simpeg.potential_fields.gravity
-        - simpeg.potential_fields.magnetics
-        - simpeg.seismic
-        - simpeg.seismic.straight_ray_tomography
-        - simpeg.utils
-        - simpeg.utils.drivers
-        - simpeg.regularization
-        - simpeg.meta
+        - ${{ module_name }}
+        - ${{ module_name }}.electromagnetics
+        - ${{ module_name }}.electromagnetics.analytics
+        - ${{ module_name }}.electromagnetics.frequency_domain
+        - ${{ module_name }}.electromagnetics.natural_source
+        - ${{ module_name }}.electromagnetics.natural_source.utils
+        - ${{ module_name }}.electromagnetics.static
+        - ${{ module_name }}.electromagnetics.static.resistivity
+        - ${{ module_name }}.electromagnetics.static.induced_polarization
+        - ${{ module_name }}.electromagnetics.static.spectral_induced_polarization
+        - ${{ module_name }}.electromagnetics.static.utils
+        - ${{ module_name }}.electromagnetics.time_domain
+        - ${{ module_name }}.electromagnetics.utils
+        - ${{ module_name }}.electromagnetics.viscous_remanent_magnetization
+        - ${{ module_name }}.flow
+        - ${{ module_name }}.flow.richards
+        - ${{ module_name }}.potential_fields
+        - ${{ module_name }}.potential_fields.gravity
+        - ${{ module_name }}.potential_fields.magnetics
+        - ${{ module_name }}.seismic
+        - ${{ module_name }}.seismic.straight_ray_tomography
+        - ${{ module_name }}.utils
+        - ${{ module_name }}.utils.drivers
+        - ${{ module_name }}.regularization
+        - ${{ module_name }}.meta
+        - ${{ module_name }}.version
+  - package_contents:
+      files:
+        - site-packages/${{ module_name }}/version.py
 
 about:
   summary: "Mira Geoscience fork of SimPEG: Simulation and Parameter Estimation in Geophysics"