Fix a bug of problem 003