Temp fix for tests.