Fixed breaking tests.