diff --git a/superxas_bec/devices/timepix/timepix.py b/superxas_bec/devices/timepix/timepix.py index 3b5ed6a..7bfeb66 100644 --- a/superxas_bec/devices/timepix/timepix.py +++ b/superxas_bec/devices/timepix/timepix.py @@ -487,9 +487,10 @@ if __name__ == "__main__": # pragma: no cover print( f"Events in energy rois {timepix.xes_data.get().signals[timepix.xes_data.name]['value'].sum()}" ) - print( - f"Total number of events: {timepix.xes_info.get().signals['tds_total_events']['value']}" - ) + events = timepix.xes_data.get().signals[ + f"{timepix.xes_data.name}_tds_total_events" + ]["value"] + print(f"Total number of events: {events}") status = timepix.complete() print("Waiting for timepix to complete.") status.wait(timeout=10)