[UPDATE] minor bug fix