Fixing integration test

reverted a bad commit and fixed integration test