diff --git a/tests/test_template.py b/tests/test_template.py
index 54d869647..66a0b0218 100644
--- a/tests/test_template.py
+++ b/tests/test_template.py
@@ -59,6 +59,7 @@ def test_build_template_from_function():
     # Test with valid name
     result = build_template_from_function("ExampleClass1", type_to_loader_dict)
 
+    assert result is not None
     assert "template" in result
     assert "description" in result
     assert "base_classes" in result
@@ -67,6 +68,7 @@ def test_build_template_from_function():
     result_with_function = build_template_from_function(
         "ExampleClass1", type_to_loader_dict, add_function=True
     )
+    assert result_with_function is not None
     assert "function" in result_with_function["base_classes"]
 
     # Test with invalid name