Fixed a bug from the last commit.