Fixed a bug with last commit.