diff --git a/src/Umbraco.Web.UI.Client/src/views/mediatypes/edit.controller.js b/src/Umbraco.Web.UI.Client/src/views/mediatypes/edit.controller.js
index ff9d412f8a..61e7b5a2dc 100644
--- a/src/Umbraco.Web.UI.Client/src/views/mediatypes/edit.controller.js
+++ b/src/Umbraco.Web.UI.Client/src/views/mediatypes/edit.controller.js
@@ -130,7 +130,7 @@
 
                             contentTypeHelper.generateModels().then(function (result) {
 
-                                if (result.success) {
+                                if (!result.lastError) {
 
                                     //re-check model status
                                     contentTypeHelper.checkModelsBuilderStatus().then(function (statusResult) {
diff --git a/src/Umbraco.Web.UI.Client/src/views/membertypes/edit.controller.js b/src/Umbraco.Web.UI.Client/src/views/membertypes/edit.controller.js
index ab42c91959..c9c32c657a 100644
--- a/src/Umbraco.Web.UI.Client/src/views/membertypes/edit.controller.js
+++ b/src/Umbraco.Web.UI.Client/src/views/membertypes/edit.controller.js
@@ -87,7 +87,7 @@
 
                             contentTypeHelper.generateModels().then(function (result) {
 
-                                if (result.success) { 
+                                if (!result.lastError) { 
 
                                     //re-check model status
                                     contentTypeHelper.checkModelsBuilderStatus().then(function (statusResult) {