Fixing bug on last commit.