diff --git a/modules/home/ai-agent/default.nix b/modules/home/ai-agent/default.nix
index 2fe4b3b..3ec1ab0 100644
--- a/modules/home/ai-agent/default.nix
+++ b/modules/home/ai-agent/default.nix
@@ -46,7 +46,8 @@ let
         paths = [ llmAgentsPkgs.codex ];
         nativeBuildInputs = [ pkgs.makeWrapper ];
         postBuild = ''
-          wrapProgram "$out/bin/codex" --add-flags '-c tui.terminal_title=[] -c tui.alternate_screen=always --enable hooks'
+          wrapProgram "$out/bin/codex" --add-flags '-c tui.terminal_title=[] -c tui.alternate_screen=always --enable hooks' \
+            --add-flags '-c tui.keymap.chat.edit_queued_message=ctrl-up -c tui.keymap.chat.prompt_stack_back=ctrl-down'
         '';
       }
     else