fixed minor bug with previous commit.