fix generation issue