Fixed addi issue in generator