repair unit tests