diff --git a/pytec/tec_qt.py b/pytec/tec_qt.py
index 4d0ddf7..f6bd77a 100755
--- a/pytec/tec_qt.py
+++ b/pytec/tec_qt.py
@@ -422,8 +422,7 @@ class MainWindow(QtWidgets.QMainWindow):
         assert self.thermostat.connected()
 
         await self.thermostat.set_ipv4(ipv4_settings)
-        await self.thermostat._client.end_session()
-        await self._on_connection_changed(False)
+        await self.bail()
 
 
 async def coro_main():