diff --git a/src/main/java/frc4388/robot/Robot.java b/src/main/java/frc4388/robot/Robot.java
index 7e548a1..3f93b48 100644
--- a/src/main/java/frc4388/robot/Robot.java
+++ b/src/main/java/frc4388/robot/Robot.java
@@ -16,7 +16,7 @@ import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;
 import edu.wpi.first.wpilibj2.command.Command;
 import edu.wpi.first.wpilibj2.command.CommandScheduler;
 import edu.wpi.first.wpilibj2.command.SubsystemBase;
-import frc4388.robot.commands.AutoBalanceTF2;
+import frc4388.robot.commands.AutoBalance;
 import frc4388.utility.RobotGyro;
 import frc4388.utility.RobotTime;
 
@@ -72,7 +72,7 @@ public class Robot extends TimedRobot {
     // autonomous chooser on the dashboard.
     m_robotContainer = new RobotContainer();
 
-    bot.setDefaultCommand(new AutoBalanceTF2(bot));
+    bot.setDefaultCommand(new AutoBalance(bot));
   }
 
   /**