fixed tests and typo from previous commit.