Add a unit test for #154