Fixed failed test

jianghaolu · May 17, 2016 · e0e5bd1 · e0e5bd1
1 parent 640df1a
commit e0e5bd1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs b/AutoRest/Generators/Python/Python/TemplateModels/MethodTemplateModel.cs
@@ -575,17 +575,17 @@ public string GetDocumentationType(IType type)
             {
                 result = string.Format(CultureInfo.InvariantCulture, "list of {0}", GetDocumentationType(listType.ElementType));
             }
-            else if (type is EnumType)
+            else if (enumType != null)
             {
-                if (type == ReturnType.Body)
+                if (enumType == ReturnType.Body)
                 {
                     if (enumType.ModelAsString)
                         result = "str";
                     else
-                        result = string.Format(CultureInfo.InvariantCulture, ":class:`{0} <{1}.models.{0}>`", type.Name, modelNamespace);
+                        result = string.Format(CultureInfo.InvariantCulture, ":class:`{0} <{1}.models.{0}>`", enumType.Name, modelNamespace);
                 }
                 else
-                    result = string.Format(CultureInfo.InvariantCulture, "str or :class:`{0} <{1}.models.{0}>`", type.Name, modelNamespace);
+                    result = string.Format(CultureInfo.InvariantCulture, "str or :class:`{0} <{1}.models.{0}>`", enumType.Name, modelNamespace);
             }
             else if (type is DictionaryType)
             {