Fix up post generation extension