diff --git a/oft/model/oftnet.py b/oft/model/oftnet.py
index d97197e..18a870a 100644
--- a/oft/model/oftnet.py
+++ b/oft/model/oftnet.py
@@ -4,7 +4,6 @@
 
 from . import resnet
 from .oft import OFT
-from .edoft import ExplicitDepthOFT
 
 class OftNet(nn.Module):