Fixing TTree filling routine