Fix forking issue when running unit tests.