[UPDATE] Small tweak