diff --git a/NanoGardener/python/modules/addTnpTree.py b/NanoGardener/python/modules/addTnpTree.py
index 036ca33a7..b87d3c275 100644
--- a/NanoGardener/python/modules/addTnpTree.py
+++ b/NanoGardener/python/modules/addTnpTree.py
@@ -90,7 +90,7 @@ def IsMatched(self, lep, trigObjCollection):
       return match
 
     def matchesPrompt(self, lep, genparts):
-        return (lep.genPartFlav == 1 and genparts[lep.genPartIdx].statusFlags & 1) or  (lep.genPartFlav == 15 and genparts[lep.genPartIdx].statusFlags & (1 << 5))
+        return (lep.genPartFlav == 1 and genparts[lep.genPartIdx].statusFlags & 1) or  (lep.genPartFlav == 15 and (genparts[lep.genPartIdx].statusFlags >> 5) & 1)
 
     def analyze(self, event):
         # Get Jet and Ele collections