diff --git a/cli/src/rust_template.rs b/cli/src/rust_template.rs
index b92c1f4..db6ce5a 100644
--- a/cli/src/rust_template.rs
+++ b/cli/src/rust_template.rs
@@ -477,10 +477,11 @@ describe("{}", () => {{
       entities: [{{
         entity: entityPda,
         components: [{{ componentId: positionComponent.programId }}],
-      }}]
-    }}, {{
+      }}],
+      args: {{
         x: 1,
         y: 1,
+      }}
     }});
     const txSign = await provider.sendAndConfirm(applySystem.transaction);
     console.log(`Applied a system. Signature: ${{txSign}}`);
@@ -491,7 +492,6 @@ describe("{}", () => {{
     );
     expect(positionAfter.x.toNumber()).to.gt(0);
   }});
-
 }});
 "#,
         name.to_upper_camel_case(),