fix the test for #158