Fixed State Bug

Fixed bug I talked about before.