whoops small fix to generate