diff --git a/recipe.yaml b/recipe.yaml
index 307498bb..c64122e7 100644
--- a/recipe.yaml
+++ b/recipe.yaml
@@ -4,6 +4,7 @@ context:
   name: "simpeg-drivers"
   version: "0.0.0.dev0"  # This will be replaced by the actual version in the build process
   python_min: "3.10"
+  module_name: ${{ name|lower|replace("-", "_") }}
 
 package:
   name: ${{ name|lower }}
@@ -58,14 +59,14 @@ requirements:
 tests:
   - python:
       imports:
+        - ${{ module_name }}
+        - ${{ module_name }}._version
         - dask
         - distributed
         - geoapps_utils
         - geoh5py
         - grid_apps
         - simpeg
-        - simpeg_drivers
-        - simpeg_drivers._version
       pip_check: true
 
   - script:
@@ -77,6 +78,10 @@ tests:
       source:
         - tests/
 
+  - package_contents:
+      files:
+        - site-packages/${{ module_name }}/_version.py
+
 about:
   license: MIT
   license_file: