Fixed bug with saving the game