Can we add a quick unit test to this?