diff --git a/lazyllm/engine/engine.py b/lazyllm/engine/engine.py
index 60d44eee..e2fa2316 100644
--- a/lazyllm/engine/engine.py
+++ b/lazyllm/engine/engine.py
@@ -465,7 +465,7 @@ def cond(x):
                     return True
         return False
 
-    return lazyllm.ifs(cond, tpath=lazyllm.TrainableModule(base_model), fpath=lazyllm.Identity())
+    return lazyllm.ActionModule(lazyllm.ifs(cond, tpath=lazyllm.TrainableModule(base_model), fpath=lazyllm.Identity()))
 
 
 @NodeConstructor.register('Constant')