From 00cfee002f1f6b2701a3724989e525613a4b4d93 Mon Sep 17 00:00:00 2001
From: Michael Hudson-Doyle <michael.hudson@canonical.com>
Date: Fri, 6 Nov 2020 09:54:30 +1300
Subject: [PATCH] trivial fixes for console_conf breakage

---
 console_conf/controllers/identity.py | 1 +
 console_conf/core.py                 | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/console_conf/controllers/identity.py b/console_conf/controllers/identity.py
index f3f8a81e..2a153c73 100644
--- a/console_conf/controllers/identity.py
+++ b/console_conf/controllers/identity.py
@@ -208,6 +208,7 @@ class IdentityController(TuiController):
         ifaces = net_model.get_all_netdevs()
         login_view = LoginView(self.opts, self.model, self, ifaces)
         login_view._w.focus_position = 2
+        return login_view
 
     def login(self):
         self.ui.set_body(self.make_login_view())
diff --git a/console_conf/core.py b/console_conf/core.py
index b7c916ec..09e63b30 100644
--- a/console_conf/core.py
+++ b/console_conf/core.py
@@ -37,6 +37,10 @@ class ConsoleConf(TuiApplication):
         "Identity",
     ]
 
+    def __init__(self, opts):
+        super().__init__(opts)
+        self.prober = Prober(opts.machine_config, self.debug_flags)
+
 
 class RecoveryChooser(TuiApplication):