Fix keymap for tree pushing in stage 4