integration tests fix