This should fix the bug seen with some registers not being allocated