diff --git a/lib/models/cell_operations.py b/lib/models/cell_operations.py
index f9e6a20..ce66794 100644
--- a/lib/models/cell_operations.py
+++ b/lib/models/cell_operations.py
@@ -284,7 +284,8 @@ class GDAS_Reduction_Cell(nn.Module):
     if self.training and drop_prob > 0.:
       X0, X1 = drop_path(X0, drop_prob), drop_path(X1, drop_prob)
 
-    X2 = self.ops2[0] (X0+X1)
+    #X2 = self.ops2[0] (X0+X1)
+    X2 = self.ops2[0] (s0)
     X3 = self.ops2[1] (s1)
     if self.training and drop_prob > 0.:
       X2, X3 = drop_path(X2, drop_prob), drop_path(X3, drop_prob)