add new changes in integration tests