Small fix for generation script