Fix unit tests for package rename