Fix generating script with new files