diff --git a/source/tests/universal/common/cases/model/utils.py b/source/tests/universal/common/cases/model/utils.py
index 1d8182e380..d5b46b6d4b 100644
--- a/source/tests/universal/common/cases/model/utils.py
+++ b/source/tests/universal/common/cases/model/utils.py
@@ -184,11 +184,12 @@ def test_forward(self) -> None:
 
         for kk in ret[0]:
             # ensure the first frame and the second frame are the same
-            np.testing.assert_allclose(
-                ret[0][kk][0],
-                ret[0][kk][1],
-                err_msg=f"compare {kk} between frame 0 and 1",
-            )
+            if ret[0][kk] is not None:
+                np.testing.assert_allclose(
+                    ret[0][kk][0],
+                    ret[0][kk][1],
+                    err_msg=f"compare {kk} between frame 0 and 1",
+                )
 
             subret = []
             for rr in ret: