diff --git a/app/common/command/build/Dist.php b/app/common/command/build/Dist.php
index 98da56b..25a2b44 100644
--- a/app/common/command/build/Dist.php
+++ b/app/common/command/build/Dist.php
@@ -264,7 +264,7 @@ class Dist extends Command
             });
 
 
-            $traverser->traverse($stmts);
+            $stmts = $traverser->traverse($stmts);
             
             foreach ($stmts as  $stmt_item) {
                 $function_stmts[] = $stmt_item;