[Update] small fix in last change