fixed last bit of test case