- Fixed minor reg alloc bug
- Made reg alloc state more robust