now with a unit test