diff --git a/src/tpl/login-initial.html b/src/tpl/login-initial.html
index 9c4629a..2fe1c9e 100644
--- a/src/tpl/login-initial.html
+++ b/src/tpl/login-initial.html
@@ -4,34 +4,33 @@
     <div class="logo"></div>
   </div><!--/logo-->
 
-  <div class="content" ng-show="state === states.IDLE">
-    <h1>Set passphrase</h1>
+  <div class="content" ng-switch on="state">
 
-    <p>This is used to protect your keypair. If you forget your passphrase, there is no way to restore your data.</p>
+    <div ng-switch-when="1">
+      <h1>Set passphrase</h1>
+      <p>This is used to protect your keypair. If you forget your passphrase, there is no way to restore your data.</p>
+      <form>
+        <div>
+          <input type="password" ng-model="passphrase" placeholder="Enter passphrase" tabindex="1" focus-me="true">
+          <input type="password" ng-model="confirmation" ng-class="{error: confirmation !== passphrase}" placeholder="Confirm passphrase" tabindex="2">
+        </div>
+        <div>
+          <button type="submit" ng-click="confirmPassphrase()" class="btn" ng-disabled="!passphrase || passphrase!==confirmation" tabindex="3">Generate keypair</button>
+        </div>
+      </form>
+    </div>
+
+    <div ng-switch-when="2">
+      <h1>Generating keypair</h1>
+      <p>Please stand by. This can take a while...</p>
+    </div>
+
+    <div ng-switch-when="4">
+      <h1>Keypair generated</h1>
+      <p>Your personal keypair has been generated. You can export it (e.g. to a USB flash drive) to setup whiteout on another computer or as a backup.</p>
+      <button ng-click="exportKeypair()" class="btn" tabindex="4">Export now</button>
+      <button ng-click="proceed()" class="btn" tabindex="5">Do it later</button>
+    </div>
 
-    <form>
-      <div>
-        <input type="password" ng-model="passphrase" placeholder="Enter passphrase" tabindex="1" focus-me="true">
-        <input type="password" ng-model="confirmation" ng-class="{error: confirmation !== passphrase}" placeholder="Confirm passphrase" tabindex="2">
-      </div>
-      <div>
-        <button type="submit" ng-click="confirmPassphrase()" class="btn" ng-disabled="!passphrase || passphrase!==confirmation" tabindex="3">Generate keypair</button>
-      </div>
-    </form>
   </div><!--/content-->
-
-  <div class="content" ng-show="state === states.PROCESSING">
-    <h1>Generating keypair</h1>
-
-    <p>Please stand by. This can take a while...</p>
-  </div><!--/content-->
-
-  <div class="content" ng-show="state === states.DONE">
-    <h1>Keypair generated</h1>
-
-    <p>Your personal keypair has been generated. You can export it (e.g. to a USB flash drive) to setup whiteout on another computer or as a backup.</p>
-    <button ng-click="exportKeypair()" class="btn" tabindex="4">Export now</button>
-    <button ng-click="proceed()" class="btn" tabindex="5">Do it later</button>
-  </div><!--/content-->
-
 </div>
\ No newline at end of file