diff --git a/benchmarks/sru.py b/benchmarks/sru.py
index 68e33b842b..1551da0c8b 100644
--- a/benchmarks/sru.py
+++ b/benchmarks/sru.py
@@ -455,12 +455,12 @@ def forward(self, u, x, bias, init=None, mask_h=None):
 
         u = u.view(length, batch, d, k_)
 
-        cur = x.new(ncols).zero_() if init is None else init
+        cur = x.new(batch, d).zero_() if init is None else init
         size = (length, batch, d*bidir) if x.dim() == 3 else (batch, d*bidir)
         bias1, bias2 = bias.split(self.d_out)
         u_ = [u.select(-1, i) for i in range(0, k_)]
         h = []
-        x_ = x if k_ == 3 else u[3]
+        x_ = x if k_ == 3 else u_[3]
         for i in range(0, length):
             u0i, u1i, u2i = u_[0][i], u_[1][i], u_[2][i]
             g1 = torch.sigmoid(u1i + bias1)