diff --git a/README.md b/README.md
index 6e8a3ca..d65c751 100644
--- a/README.md
+++ b/README.md
@@ -9,6 +9,7 @@ data = dataset[0]
 
 import torch.nn.functional as F
 from torch_geometric.nn import GCNConv, global_mean_pool
+from eixgnn.eixgnn import EiXGNN
 
 class GCN(torch.nn.Module):
     def __init__(self):