diff --git a/wallet-web/components/undelegate/NodeUndelegation.tsx b/wallet-web/components/undelegate/NodeUndelegation.tsx
index c04aa1464d..34666e48e0 100644
--- a/wallet-web/components/undelegate/NodeUndelegation.tsx
+++ b/wallet-web/components/undelegate/NodeUndelegation.tsx
@@ -65,7 +65,7 @@ const UndelegateFromNode = () => {
     }
 
     // we haven't clicked undelegate button yet
-    if (!undelegationStarted) {
+    if (!isLoading) {
       return (
         <React.Fragment>
           <NodeTypeChooser nodeType={nodeType} setNodeType={setNodeType} />
@@ -95,9 +95,9 @@ const UndelegateFromNode = () => {
         <Paper className={classes.paper}>
           <ExecFeeNotice name={'undelegating stake'} />
           <Typography
-            component='h1'
-            variant='h4'
-            align='center'
+            component="h1"
+            variant="h4"
+            align="center"
             className={classes.wrapper}
           >
             Undelegate stake from {nodeType}