Fixed some parts of the register allocator