diff --git a/explaining_framework/metric/robust.py b/explaining_framework/metric/robust.py
index 4e74654..b4348d9 100644
--- a/explaining_framework/metric/robust.py
+++ b/explaining_framework/metric/robust.py
@@ -2,10 +2,10 @@ import copy
 
 import torch
 import torch.nn.functional as F
+from torch.nn import CrossEntropyLoss, MSELoss
 from torch_geometric.explain.explanation import Explanation
 from torch_geometric.graphgym.config import cfg
 from torch_geometric.utils import add_random_edge, dropout_edge, dropout_node
-from troch.nn import CrossEntropyLoss, MSELoss
 
 from explaining_framework.metric.base import Metric