Fix broken package name in recent refactor