Fixed a bug of last commit.