diff --git a/lib/route-helper.js b/lib/route-helper.js
index 80bf9a8..ce65cc2 100644
--- a/lib/route-helper.js
+++ b/lib/route-helper.js
@@ -88,10 +88,7 @@ var routeHelper = module.exports = {
       if (firstReturn.type === 'object') {
         firstReturn.type = classDef.name;
       } else if (firstReturn.type === 'array') {
-        firstReturn.type = 'array';
-        firstReturn.items = {
-          '$ref': classDef.name
-        };
+        firstReturn.type = [classDef.name];
       }
     }