Integration tests fix