Fix integration test regression.

* Related commit: 7a1c4429a58e2c6b1f3424e74639dd0639d67a50.