Updating integration tests