Fix test for #322