Fix auto gen bug