1 files changed, 5 insertions, 8 deletions
diff --git a/templates/generate_system.m b/templates/generate_system.m
index c4dadc2..01e1b59 100644
--- a/templates/generate_system.m
+++ b/templates/generate_system.m
@@ -8,12 +8,9 @@
 
 function [A, B] = generate_system(Ac, Bc, params)
     % YOUR CODE HERE
-    % Ts = 600;
-    sysc = ss(Ac, Bc, [], []);
-    Ts = params.model.TimeStep;
-    sysd = c2d(sysc,Ts);
-    % A = eye(params.model.nx) + Ac * Ts;
-    % B = Bc * Ts;
-    A = sysd.A;
-    B = sysd.B;
+    t=params.model.TimeStep;
+    sys_c=ss(Ac,Bc,[],[]);
+    sys_d = c2d(sys_c, t, 'zoh');
+    A = sys_d.A;
+    B = sys_d.B;
 end
 \ No newline at end of file