diff --git a/model/mask_regression.py b/model/mask_regression.py
index 8553f10..ed53962 100644
--- a/model/mask_regression.py
+++ b/model/mask_regression.py
@@ -3,7 +3,7 @@
 import torch.nn.functional as F
 from utils.bilinear import *
 from .sync_batchnorm import SynchronizedBatchNorm2d
-from .norm_module import AdaptiveBatchNorm2
+from .norm_module import AdaptiveBatchNorm2d
 import pickle
 import numpy as np
 from torch.nn import Parameter