Solving test case that was failing