Fix logic in how metric states referencing is handled in MetricCollection

Fix logic in how metric states referencing is handled in MetricCollection #11008