Add a unit test for #226