diff --git a/modules/source/06_spatial/spatial_dev.py b/modules/source/06_spatial/spatial_dev.py
index ebf9541a..79d00396 100644
--- a/modules/source/06_spatial/spatial_dev.py
+++ b/modules/source/06_spatial/spatial_dev.py
@@ -821,9 +821,17 @@ def test_module_conv2d_tensor_compatibility():
 ## 🤖 AUTO TESTING
 """
 
+# %%
+# Run the integration test
+test_module_conv2d_tensor_compatibility()
+
+# %% [markdown]
+"""
+## 🤖 AUTO TESTING
+"""
+
 # %%
 if __name__ == "__main__":
-    test_module_conv2d_tensor_compatibility()
     from tito.tools.testing import run_module_tests_auto
     
     # Automatically discover and run all tests in this module
diff --git a/modules/source/08_dataloader/dataloader_dev.py b/modules/source/08_dataloader/dataloader_dev.py
index 1be07062..00201e7e 100644
--- a/modules/source/08_dataloader/dataloader_dev.py
+++ b/modules/source/08_dataloader/dataloader_dev.py
@@ -1090,6 +1090,15 @@ def test_module_dataloader_tensor_yield():
 ## 🤖 AUTO TESTING
 """
 
+# %%
+# Run the integration test
+test_module_dataloader_tensor_yield()
+
+# %% [markdown]
+"""
+## 🤖 AUTO TESTING
+"""
+
 # %%
 if __name__ == "__main__":
     from tito.tools.testing import run_module_tests_auto