Fix build error in integration test
Merge pull request #1631 from lavalamp/fix

Fix build error in integration test