Fix error in generation script