On the choice of the density matrix in the stochastic TMRG