Fixes unit tests.