diff --git a/agentops/cli.py b/agentops/cli.py
index fdfad396..a79de122 100644
--- a/agentops/cli.py
+++ b/agentops/cli.py
@@ -32,6 +32,6 @@ def cli():
         if args.branch_name:
             fetch_time_travel_id(args.branch_name)
         if args.on:
-            set_time_travel_active_state("on")
+            set_time_travel_active_state(True)
         if args.off:
-            set_time_travel_active_state("off")
+            set_time_travel_active_state(False)