Revert "Update integration test"

This reverts commit 134879b08eb0f7a892df6d8976e6991861b47758.