Could you add a unit test for this?