Fixed missing updateStats call