revising unit tests