From a4f3be1ad55e9937ac95c0dc970b794c877b65b5 Mon Sep 17 00:00:00 2001
From: appel_c <christian.appel@psi.ch>
Date: Mon, 2 Mar 2026 09:20:17 +0100
Subject: [PATCH] refator: fix patch_dual_pvs

---
 ophyd_devices/tests/utils.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/ophyd_devices/tests/utils.py b/ophyd_devices/tests/utils.py
index 3700e77..2c6dbfd 100644
--- a/ophyd_devices/tests/utils.py
+++ b/ophyd_devices/tests/utils.py
@@ -76,7 +76,12 @@ def patched_device(
 def patch_dual_pvs(device):
     """Patch dual PVs"""
     patch_functions_required_for_connection(device)
-    device.wait_for_connection(all_signals=True)
+    try:
+        device.wait_for_connection(all_signals=True)
+    except TypeError:
+        # Maybe no all_signals in kwargs,
+        device.wait_for_connection()
+
     for walk in device.walk_signals():
         if not hasattr(walk.item, "_read_pv"):
             continue