diff --git a/system_setup/client/controllers/identity.py b/system_setup/client/controllers/identity.py
index 39b3e3b6..48364b53 100644
--- a/system_setup/client/controllers/identity.py
+++ b/system_setup/client/controllers/identity.py
@@ -15,14 +15,16 @@
 
 import logging
 
-from subiquity.client.controllers import IdentityController
+from subiquity.client.controller import SubiquityTuiController
 from subiquity.common.types import IdentityData
 from system_setup.ui.views import WSLIdentityView
 
 log = logging.getLogger('system_setup.client.controllers.identity')
 
 
-class WSLIdentityController(IdentityController):
+class WSLIdentityController(SubiquityTuiController):
+
+    endpoint_name = 'identity'
 
     async def make_ui(self):
         data = await self.endpoint.GET()
@@ -37,6 +39,9 @@ class WSLIdentityController(IdentityController):
                 crypted_password=self.answers['password'])
             self.done(identity)
 
+    def cancel(self):
+        self.app.prev_screen()
+
     def done(self, identity_data):
         log.debug(
             "IdentityController.done next_screen user_spec=%s",