Can you add an assert with the previous entity count here, to make the difference before-after more obvious?