Fix warning in problem 17