Fix assignments of registers for jumps