diff --git a/graph_dit/metrics/molecular_metrics_train.py b/graph_dit/metrics/molecular_metrics_train.py
index f141fb1..7200a9d 100644
--- a/graph_dit/metrics/molecular_metrics_train.py
+++ b/graph_dit/metrics/molecular_metrics_train.py
@@ -85,7 +85,7 @@ class EdgeMetricsCE(MetricCollection):
         ce_SI = SingleCE(1)
         ce_DO = DoubleCE(2)
         ce_TR = TripleCE(3)
-        super().__init__([ce_no_bond, ce_SI, ce_DO, ce_TR])
+        super().__init__([ce_no_bond, ce_SI])
 
 class AtomMetricsCE(MetricCollection):
     def __init__(self, active_atoms):