Fixed bug related to my last commit.