cleaned up the unit test