Update some integration tests