[examples] Tweak multistage example