Add type in GeneratorCommand