diff --git a/main.py b/main.py
index f666d448..e51386dc 100644
--- a/main.py
+++ b/main.py
@@ -157,7 +157,7 @@
                 agent.add(state, action, reward, next_state, mask, 1)
             else:
                 agent.add(state, action, reward, next_state, mask, 0)
-        for (state, action, rewards, next_state, mask, _) in real_buffer:
+        for (state, action, reward, next_state, mask, _) in real_buffer:
             if cost > 0:
                 real_data.push(state, action, reward, next_state, mask, 1)
             else: