Fix the issue that resulted in wrong answer for the sample data

Fix the issue that resulted in wrong answer for the sample data #36