fix bug in last commit.