Fixed up register re allocation