diff --git a/modules/source/04_layers/layers_dev.py b/modules/source/04_layers/layers_dev.py
index f7df7d05..9823b1f5 100644
--- a/modules/source/04_layers/layers_dev.py
+++ b/modules/source/04_layers/layers_dev.py
@@ -655,8 +655,8 @@ def test_layer_tensor_integration():
     assert np.array_equal(output_tensor.data, np.array([2, 4, 6])), "Output data is incorrect"
     print("✅ Integration Test Passed: Layer correctly processed Tensor.")
 
-if __name__ == "__main__":
-    test_layer_tensor_integration()
+# Run the integration test
+test_layer_tensor_integration()
 
 
 # %% [markdown]