diff --git a/reco/KF/CbmKFParticleFinder.cxx b/reco/KF/CbmKFParticleFinder.cxx
index a65f3c9a6e3bd4a23f7bbc0310712f47cb860536..0a74bf63e46db0f67f12293f98a517008e4c60ef 100644
--- a/reco/KF/CbmKFParticleFinder.cxx
+++ b/reco/KF/CbmKFParticleFinder.cxx
@@ -553,7 +553,7 @@ void CbmKFParticleFinder::FillKFPTrackVector(
     tracks->SetQ(q, iTr);
     tracks->SetNPixelHits(vRTracks[iTr].GetNofMvdHits(), iTr);
 
-    if (fPVFindMode == 0) {
+    if (fPVFindMode == 0 || fPVFindMode == 3) {
       if (vChiToPrimVtx[iTr] < 3)
         tracks->SetPVIndex(0, iTr);
       else