Fix generate button