Still trying to write a unit test