1 files changed, 1 insertions, 4 deletions
diff --git a/utils/plot-muons b/utils/plot-muons
index 0646e2e..69dda87 100755
--- a/utils/plot-muons
+++ b/utils/plot-muons
@@ -79,10 +79,7 @@ if __name__ == '__main__':
     ev_mc = ev_mc.reset_index()
 
     # Set all prompt events in the MC to be muons
-    #ev_mc.loc[ev_mc.prompt,'muon'] = True
-
-    # FIXME: TESTING
-    ev_mc.loc[ev_mc.prompt & (ev_mc.id == 22) & (ev_mc.r_psup > 0.9),'muon'] = True
+    ev_mc.loc[ev_mc.prompt & ev_mc.filename.str.contains("cosmic"),'muon'] = True
 
     # First, do basic data cleaning which is done for all events.
     ev = ev[ev.signal | ev.muon]