diff --git a/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.controller.js b/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.controller.js
index 63672879b0..eac4fab8a7 100644
--- a/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.controller.js
+++ b/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.controller.js
@@ -19,6 +19,7 @@
             vm.variantVersions = [];
             vm.diff = null;
             vm.currentVersion = null;
+            vm.rollbackButtonDisabled = true;
 
             // find the current version for invariant nodes
             if($scope.model.node.variants.length === 1) {
@@ -72,10 +73,12 @@
                         vm.previousVersion = data;
                         vm.previousVersion.versionId = version.versionId;
                         createDiff(vm.currentVersion, vm.previousVersion);
+                        vm.rollbackButtonDisabled = false;
                     });
 
             } else {
                 vm.diff = null;
+                vm.rollbackButtonDisabled = true;
             }
         }
 
diff --git a/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.html b/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.html
index 5faba3a3ef..6758f1df9b 100644
--- a/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.html
+++ b/src/Umbraco.Web.UI.Client/src/views/common/infiniteeditors/rollback/rollback.html
@@ -93,6 +93,7 @@
                     button-style="success"
                     state="vm.rollbackButtonState"
                     label-key="actions_rollback"
+                    disabled="vm.rollbackButtonDisabled"
                     action="vm.rollback()">
                 </umb-button>
             </umb-editor-footer-content-right>