diff --git a/rllib/examples/cartpole_lstm.py b/rllib/examples/cartpole_lstm.py
index 1f5cb0128fd45..ce01295380f98 100644
--- a/rllib/examples/cartpole_lstm.py
+++ b/rllib/examples/cartpole_lstm.py
@@ -105,7 +105,8 @@
     # >>
     # >> while True:
     # >>     a, state_out, _ = algo.compute_single_action(
-    # ..         obs, state, prev_a, prev_r)
+    # ..         obs, state, prev_action=prev_a, prev_reward=prev_r
+    # ..     )
     # >>     obs, reward, done, truncated, _ = env.step(a)
     # >>     if done:
     # >>         obs, info = env.reset()