Fixing one of the integration tests.