Fix gen jet fetching
Working version
inherit from the base class in a PyROOT friendly way