Fix shoddy test case