Fix test case from previous commit