1 files changed, 4 insertions, 0 deletions
diff --git a/utils/plot.py b/utils/plot.py
index b74e14a..7abb7e3 100755
--- a/utils/plot.py
+++ b/utils/plot.py
@@ -65,6 +65,10 @@ if __name__ == '__main__':
         for event in data['data']:
             # get the particle ID
             id = event['mctk'][-1]['id']
+
+            if id not in event['ev'][0]['fit']:
+                continue
+
             mass = SNOMAN_MASS[id]
             # for some reason it's the *second* track which seems to contain the
             # initial energy