Small modification to how GameStates work