Revert "Fixing integration test"

This reverts commit 3b71161a389276b57a25b5a5bd213a37a7a678b7.