Fixed issue in TB simulation.