fixed up unit test