diff --git a/Reconstruction/tauRec/python/tauRecFlags.py b/Reconstruction/tauRec/python/tauRecFlags.py
index 4921378e371b794b115f799fbc030184177fa71f..586f9a66f48cd6327dd1974ed9c73c66050d7180 100644
--- a/Reconstruction/tauRec/python/tauRecFlags.py
+++ b/Reconstruction/tauRec/python/tauRecFlags.py
@@ -107,7 +107,7 @@ class tauRecRNNTrackClassificationConfig(JobProperty):
     """
     statusOn=True
     allowedTypes=['string']
-    StoredValue="RNNTrackClassifier.2021-04-28_14-53-11_90_15_30.json"
+    StoredValue="RNNTrackClassifier_2021-07-19_14-25-14_90_25_30.json"
 
 class tauRecDecayModeNNClassifierConfig(JobProperty):
     """Config file for TauDecayModeNNClassifier
diff --git a/Reconstruction/tauRecTools/Root/TauTrackRNNClassifier.cxx b/Reconstruction/tauRecTools/Root/TauTrackRNNClassifier.cxx
index ddf8464cb07cf1072373539aadc1fcfe904d057e..a79a9906a850ccbd8a03b7877aaea58570c46b0a 100644
--- a/Reconstruction/tauRecTools/Root/TauTrackRNNClassifier.cxx
+++ b/Reconstruction/tauRecTools/Root/TauTrackRNNClassifier.cxx
@@ -259,8 +259,8 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
   }
 
   valueMap["log(trackPt)"] = std::vector<double>(n_timeSteps);
-  valueMap["log(tauPtIntermediateAxis)"] = std::vector<double>(n_timeSteps);
-  valueMap["(trackPt/tauPtIntermediateAxis[0])"] = std::vector<double>(n_timeSteps);
+  valueMap["log(jetSeedPt)"] = std::vector<double>(n_timeSteps);
+  valueMap["trackPt/tauPtIntermediateAxis"] = std::vector<double>(n_timeSteps);
   valueMap["trackEta"] = std::vector<double>(n_timeSteps);
   valueMap["z0sinthetaTJVA"] = std::vector<double>(n_timeSteps);
   valueMap["z0sinthetaSigTJVA"] = std::vector<double>(n_timeSteps);
@@ -275,7 +275,7 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
   valueMap["numberOfPixelSharedHits"] = std::vector<double>(n_timeSteps);
   valueMap["numberOfSCTSharedHits"] = std::vector<double>(n_timeSteps);
   valueMap["numberOfTRTHits"] = std::vector<double>(n_timeSteps);
-  valueMap["eProbabilityNNorHT"] = std::vector<double>(n_timeSteps);
+  valueMap["eProbabilityHT"] = std::vector<double>(n_timeSteps);
   valueMap["nPixHits"] = std::vector<double>(n_timeSteps);
   valueMap["nSCTHits"] = std::vector<double>(n_timeSteps);
   valueMap["dz0_TV_PV0"] = std::vector<double>(n_timeSteps);
@@ -286,7 +286,7 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
   valueMap["charge"] = std::vector<double>(n_timeSteps);
 
   // tau variable
-  double log_ptIntermediateAxis = std::log( xTau.ptIntermediateAxis() );
+  double log_ptJetSeed = std::log( xTau.ptJetSeed() );
 
   // vertex variables
   double dz0_TV_PV0 = 0., sumpt_TV = 0., sumpt2_TV = 0., sumpt_PV0 = 0., sumpt2_PV0 = 0.;
@@ -310,7 +310,6 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
 
   // track variables
   unsigned int i = 0;
-  static const SG::AuxElement::ConstAccessor<float> acc_eProbabilityNN("eProbabilityNN");
 
   for(xAOD::TauTrack* xTrack : vTracks)
     {
@@ -324,15 +323,11 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
       uint8_t nSCTSharedHits = 0; ATH_CHECK( xTrackParticle->summaryValue(nSCTSharedHits, xAOD::numberOfSCTSharedHits) );
       uint8_t nSCTDeadSensors = 0; ATH_CHECK( xTrackParticle->summaryValue(nSCTDeadSensors, xAOD::numberOfSCTDeadSensors) );
       uint8_t nTRTHits = 0; ATH_CHECK( xTrackParticle->summaryValue(nTRTHits, xAOD::numberOfTRTHits) );
-      // currently not used, to be considered for future trainings
-      //uint8_t nTRTHighThresholdHits = 0; ATH_CHECK( xTrackParticle->summaryValue(nTRTHighThresholdHits, xAOD::numberOfTRTHighThresholdHits) );
       float eProbabilityHT; ATH_CHECK( xTrackParticle->summaryValue( eProbabilityHT, xAOD::eProbabilityHT) );
-      float eProbabilityNN = acc_eProbabilityNN(*xTrackParticle);
-      float eProbabilityNNorHT = (xTrackParticle->pt()>2000.) ? eProbabilityNN : eProbabilityHT;
 
       valueMap["log(trackPt)"][i] = std::log( xTrackParticle->pt() );
-      valueMap["log(tauPtIntermediateAxis)"][i] = log_ptIntermediateAxis;
-      valueMap["(trackPt/tauPtIntermediateAxis[0])"][i] = xTrackParticle->pt()/xTau.ptIntermediateAxis();
+      valueMap["log(jetSeedPt)"][i] = log_ptJetSeed;
+      valueMap["trackPt/tauPtIntermediateAxis"][i] = xTrackParticle->pt()/xTau.ptIntermediateAxis();
       valueMap["trackEta"][i] = xTrackParticle->eta();
       valueMap["z0sinthetaTJVA"][i] = xTrack->z0sinthetaTJVA();
       valueMap["z0sinthetaSigTJVA"][i] = xTrack->z0sinthetaSigTJVA();
@@ -347,7 +342,7 @@ StatusCode TrackRNN::calulateVars(const std::vector<xAOD::TauTrack*>& vTracks,
       valueMap["numberOfPixelSharedHits"][i] = (double) nPixelSharedHits;
       valueMap["numberOfSCTSharedHits"][i] = (double) nSCTSharedHits;
       valueMap["numberOfTRTHits"][i] = (double) nTRTHits;
-      valueMap["eProbabilityNNorHT"][i] = eProbabilityNNorHT;
+      valueMap["eProbabilityHT"][i] = eProbabilityHT;
       valueMap["nPixHits"][i] = (double) (nPixelHits + nPixelDeadSensors);
       valueMap["nSCTHits"][i] = (double) (nSCTHits + nSCTDeadSensors);
       valueMap["dz0_TV_PV0"][i] = dz0_TV_PV0;
diff --git a/Tools/PROCTools/data/master_q431_AOD_digest_500events.ref b/Tools/PROCTools/data/master_q431_AOD_digest_500events.ref
index d15ded161df89a473cd1aec9c530a664648834a9..851baef8f07037413dfd99406990530c3a72b4a7 100644
--- a/Tools/PROCTools/data/master_q431_AOD_digest_500events.ref
+++ b/Tools/PROCTools/data/master_q431_AOD_digest_500events.ref
@@ -420,14 +420,14 @@
       330470  1183932465         456         541           3          11           0          11          17           0          17
       330470  1183933264         514         450           2           4           0           4          10           0          10
       330470  1183934061         556         608           1          15           0          15          31           0          31
-      330470  1183934188         559         638           0          11           0          11          28           0          28
+      330470  1183934188         559         640           0          11           0          11          28           0          28
       330470  1183935176         505         635           1           7           0           7          21           0          21
       330470  1183935243         406         422           3          15           0          15          26           0          26
       330470  1183936048         505         442           0          13           0          13          24           0          24
       330470  1183937915         358         330           0           5           0           5           9           0           9
       330470  1183937927         530         536           2           8           0           8          24           0          24
       330470  1183938712         427         382           1           2           0           2          20           0          20
-      330470  1183938824         345         395           2           3           0           3          11           0          11
+      330470  1183938824         345         394           2           3           0           3          11           0          11
       330470  1183939219         533         500           2           9           0           9          16           0          16
       330470  1183939285         273         161           0           5           0           5           8           0           8
       330470  1183939881         561         508           3           6           0           6          22           0          22