Fixed a introduced bug from the last commit.