Set correct main for package