Updating the tests in integration/