fix test errors in integration test