Fix unit tests after rename.