diff --git a/examples/model.py b/examples/model.py
index 81aed1a..de2c285 100644
--- a/examples/model.py
+++ b/examples/model.py
@@ -7,7 +7,6 @@
 from torch_geometric.data import HeteroData
 from torch_geometric.nn import MLP
 from torch_geometric.typing import NodeType
-from hybridgnn.nn import 
 from relbench.modeling.nn import HeteroEncoder, HeteroGraphSAGE, HeteroTemporalEncoder