Fix for the SetDifficulty function.