[Integration] Update tests