Fix package name after refactor