Fix regression breaking generate