You've created an infinite loop. It only triggers when your code reaches the second sample test, so when your code returns the correct answer for the first test, your code will time out.