fix bug in last commit