Fix unittest tests after previous commit.
2 files changed