[TASK] Small tweak