fix for tests
fix