diff --git a/pytec/tec_qt.py b/pytec/tec_qt.py
index 0650ae6..25a8081 100755
--- a/pytec/tec_qt.py
+++ b/pytec/tec_qt.py
@@ -418,8 +418,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():