update generator to use correct pretty printer