From b2c32b80e36589cedf62388e8c430d44c9ff0ad3 Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas.kluyver@xfel.eu>
Date: Wed, 29 May 2024 12:04:44 +0100
Subject: [PATCH] Remove keys_for_source in gotthard2lib as well

---
 src/cal_tools/gotthard2/gotthard2lib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cal_tools/gotthard2/gotthard2lib.py b/src/cal_tools/gotthard2/gotthard2lib.py
index 061bea183..8b2481833 100644
--- a/src/cal_tools/gotthard2/gotthard2lib.py
+++ b/src/cal_tools/gotthard2/gotthard2lib.py
@@ -42,7 +42,7 @@ class Gotthard2Ctrl():
 
     def get_single_photon(self):
         """Get single photon for gotthard2."""
-        if "singlePhoton.value" in self.run_dc.keys_for_source(self.ctrl_src):
+        if "singlePhoton" in self.run_dc[self.ctrl_src]:
             return bool(
                 self.run_dc[self.ctrl_src, "singlePhoton"].as_single_value())
 
-- 
GitLab