do you think you could add a unit test for this?