cleaning up unit tests