Trying to write a unit test