diff --git a/libnymea-modbus/tools/generate-connection.py b/libnymea-modbus/tools/generate-connection.py
index 3e2a251..5b1ae97 100644
--- a/libnymea-modbus/tools/generate-connection.py
+++ b/libnymea-modbus/tools/generate-connection.py
@@ -589,6 +589,12 @@ def writeRtuSourceFile():
     writeLine(sourceFile)
     writeLine(sourceFile, '        evaluateReachableState();')
     writeLine(sourceFile, '    });')
+    writeLine(sourceFile)
+    writeLine(sourceFile, '    if (m_modbusRtuMaster->connected()) {')
+    writeLine(sourceFile, '        testReachability();')
+    writeLine(sourceFile, '    } else {')
+    writeLine(sourceFile, '        evaluateReachableState();')
+    writeLine(sourceFile, '    }')
     writeLine(sourceFile, '}')
     writeLine(sourceFile)