I'm trying to put a unit test together