Fix the "new difficulty" option after winning