Fix test case for 0.8.x