diff --git a/templates/endpoint/basename.controller.js b/templates/endpoint/basename.controller.js index 24c3946c3..d5b93ef7a 100644 --- a/templates/endpoint/basename.controller.js +++ b/templates/endpoint/basename.controller.js @@ -24,13 +24,15 @@ function respondWithResult(res, statusCode) { function saveUpdates(updates) { return function(entity) { - <%_ if(filters.mongooseModels) { -%> - var updated = _.merge(entity, updates); - return updated.save(); - <%_ } -%> - <%_ if(filters.sequelizeModels) { -%> - return entity.updateAttributes(updates); - <%_ } -%> + if (entity) { + <%_ if(filters.mongooseModels) { -%> + var updated = _.merge(entity, updates); + return updated.save(); + <%_ } -%> + <%_ if(filters.sequelizeModels) { -%> + return entity.updateAttributes(updates); + <%_ } -%> + } }; }