diff --git a/src/tests/test_agipd_kernels.py b/src/tests/test_agipd_kernels.py
index 4a5aa10db181738274ebbd2c21eb90e44c2b39c1..dd1fef68eb5d656d6bcedad283fb62437e5bf8d1 100644
--- a/src/tests/test_agipd_kernels.py
+++ b/src/tests/test_agipd_kernels.py
@@ -37,6 +37,7 @@ kernel_runner = agipd_gpu.AgipdGpuRunner(
     constant_memory_cells=memory_cells,
     input_data_dtype=input_dtype,
     output_data_dtype=output_dtype,
+    output_gain_map=True,
 )
 
 
diff --git a/src/tests/test_calcat_utils.py b/src/tests/test_calcat_utils.py
index 537bddd9d5efdf4b003c4b55f6b459097ace8254..69641d2cc380e60a9d2a62ede006ec728ca20dd2 100644
--- a/src/tests/test_calcat_utils.py
+++ b/src/tests/test_calcat_utils.py
@@ -70,6 +70,9 @@ class DummyDsscDevice:
     device_class_schema = Schema()
     managed_keys = []
 
+    def log_status_info(self, msg):
+        print(msg)
+
     @staticmethod
     def expectedParameters(expected):
         # super(DummyDsscDevice, DummyDsscDevice).expectedParameters(expected)