Remove extra yield from test case.