1 files changed, 10 insertions, 0 deletions
diff --git a/src/RobotRace.java b/src/RobotRace.java
index ee8abb5..622f82b 100644
--- a/src/RobotRace.java
+++ b/src/RobotRace.java
@@ -372,6 +372,16 @@ public class RobotRace extends Base {
      */
     private void calculateRobotSpeedAndLocation() {
         double current_t = gs.tAnim;
+        // on reset, position the robots on the begin
+        if (current_t < last_speed_update) {
+            last_speed_update = 0;
+            last_t = 0;
+            for (Robot robot : robots) {
+                robot.setSpeed(0);
+                robot.resetPosition();
+            }
+        }
+
         double last_speed_update_t_diff = current_t - last_speed_update;
         if (last_speed_update_t_diff >= SPEED_RECALC_INTERVAL) {
             for (Robot robot : robots) {