Fix test case for #2