Fixed breaking test.