Integration tests updated