Small fix for script generation