diff --git a/examples/agents/keyboard_agent.py b/examples/agents/keyboard_agent.py
index d4e9d5cf1..77c29b9da 100644
--- a/examples/agents/keyboard_agent.py
+++ b/examples/agents/keyboard_agent.py
@@ -34,8 +34,8 @@ def key_release(key, mod):
         human_agent_action = 0
 
 env.render()
-env.viewer.window.on_key_press = key_press
-env.viewer.window.on_key_release = key_release
+env.unwrapped.viewer.window.on_key_press = key_press
+env.unwrapped.viewer.window.on_key_release = key_release
 
 def rollout(env):
     global human_agent_action, human_wants_restart, human_sets_pause