Fix broken reference in unit tests.