#include <asm/module.h>
 #include <asm/msa.h>
 #include <asm/ptrace.h>
+#include <asm/regdef.h>
 #include <asm/sections.h>
 #include <asm/siginfo.h>
 #include <asm/tlbdebug.h>
                unsigned long jump_mask = ~((1 << 28) - 1);
 #endif
                u32 *buf = (u32 *)(ebase + 0x200);
-               unsigned int k0 = 26;
                if ((handler & jump_mask) == ((ebase + 0x200) & jump_mask)) {
                        uasm_i_j(&buf, handler & ~jump_mask);
                        uasm_i_nop(&buf);
                } else {
-                       UASM_i_LA(&buf, k0, handler);
-                       uasm_i_jr(&buf, k0);
+                       UASM_i_LA(&buf, GPR_K0, handler);
+                       uasm_i_jr(&buf, GPR_K0);
                        uasm_i_nop(&buf);
                }
                local_flush_icache_range(ebase + 0x200, (unsigned long)buf);