fix up test failure caused by package move