Made register allocator work again