diff --git a/eixgnn/eixgnn.py b/eixgnn/eixgnn.py
index 4a860b7..bb46170 100644
--- a/eixgnn/eixgnn.py
+++ b/eixgnn/eixgnn.py
@@ -265,7 +265,3 @@ class EiXGNN(ExplainerAlgorithm):
             sub_pred = self.model(x=x, edge_index=edge_index_sub)
             node_importance[node_index] = torch.norm(pred - sub_pred, p=1)
         return node_importance.to(x.device)
-
-
-if __name__ == "__main__":
-