[UPDATE] small fix