diff --git a/lib/grape-swagger.rb b/lib/grape-swagger.rb
index a51bb65..20c855a 100644
--- a/lib/grape-swagger.rb
+++ b/lib/grape-swagger.rb
@@ -159,8 +159,20 @@ module Grape
                   if operation[:parameters].any? { | param | param[:type] == "File" }
                     operation.merge!(:consumes => [ "multipart/form-data" ])
                   end
-                  operation.merge!(:type => parse_entity_name(route.route_entity)) if route.route_entity
                   operation.merge!(:responseMessages => http_codes) unless http_codes.empty?
+
+                  if route.route_entity
+                    type = parse_entity_name(route.route_entity)
+                    if route.instance_variable_get(:@options)[:is_array]
+                      operation.merge!({
+                        "type" => "array",
+                        "items" => generate_typeref(type)
+                      })
+                    else
+                      operation.merge!("type" => type)
+                    end
+                  end
+
                   operation
                 end.compact
                 apis << {