diff --git a/ai.py b/ai.py
index cf6587e..d070601 100644
--- a/ai.py
+++ b/ai.py
@@ -33,7 +33,7 @@ def doop(er1, er2):
     
         p1 = op(erS[:sz], erL[:sz])
         erSrem = erS[sz:]
-        erLrem = erS[sz:]
+        erLrem = erL[sz:]
         p2 = op(erSrem, h.zeros(erSrem.shape))
         p3 = op(h.zeros(erLrem.shape), erLrem)
         return torch.cat((p1,p2,p3), dim=0)