Fixed minor bug on previous commit.