Fix handling of imm instruction.