tmp fix for tests