make unit tests pass