Replace re-running all tests with some unit tests