Trying to fix gamestate