diff --git a/lib/configuration/hooks/studio/index.js b/lib/configuration/hooks/studio/index.js
index 4c9433b949..82e9ab545d 100644
--- a/lib/configuration/hooks/studio/index.js
+++ b/lib/configuration/hooks/studio/index.js
@@ -326,11 +326,11 @@ module.exports = function (strapi) {
           return model;
         }),
         api: _.mapValues(_.cloneDeep(strapi.api), function (api) {
-          return _.mapValues(api.models, function (model) {
+          _.forEach(api.models, function (model) {
             model.attributes = _.omit(model.attributes, _.isFunction);
-
-            return model;
           });
+
+          return api;
         }),
         templates: {}
       };