Correcting the test case, now green