Backport integration fixes

@strzalek