Updates integration tests with small update