diff --git a/lib/models/connect.py b/lib/models/connect.py
index 86be5e8..77edf19 100644
--- a/lib/models/connect.py
+++ b/lib/models/connect.py
@@ -2,6 +2,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
+from .dcn.deform_conv import DeformConv
 
 class Corr_Up(nn.Module):
     """