updates to unit test