[POLISH] small code refactor for inline distractor demo