diff --git a/network/resnet101_asp_oc.py b/network/resnet101_asp_oc.py
index 393b23d..0e89b28 100755
--- a/network/resnet101_asp_oc.py
+++ b/network/resnet101_asp_oc.py
@@ -64,7 +64,9 @@ def __init__(self, block, layers, num_classes):
 
         # extra added layers
         self.context = nn.Sequential(
-                ASP_OC_Module(2048, 512),
+                nn.Conv2d(2048, 512, kernel_size=3, stride=1, padding=1),
+                InPlaceABNSync(512),
+                ASP_OC_Module(512, 512)
                 )
         self.cls = nn.Conv2d(512, num_classes, kernel_size=1, stride=1, padding=0, bias=True)
         self.dsn = nn.Sequential(