Fixed bug from the previous commit.