Fixed tests for the previous commit.