Fix exception found in integration tests