restore unit tests