diff --git a/esbn_pytorch/esbn_pytorch.py b/esbn_pytorch/esbn_pytorch.py
index df4e33f..d4b16bf 100644
--- a/esbn_pytorch/esbn_pytorch.py
+++ b/esbn_pytorch/esbn_pytorch.py
@@ -79,7 +79,7 @@ def forward(self, images):
 
                 # concat confidence to memory keys
                 # then weighted sum of all memory keys by attention of memory values
-                kr = g * (wk * torch.cat((Mk, ck), dim = -1)).sum(dim = 1)
+                kx = g * (wk * torch.cat((Mk, ck), dim = -1)).sum(dim = 1)
 
             kw, z = map_fn(rearrange, 'b d -> b () d')(kw, z)
             Mk = safe_cat(Mk, kw, dim = 1)
diff --git a/setup.py b/setup.py
index e040be0..7f7f8c5 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'esbn-pytorch',
   packages = find_packages(),
-  version = '0.0.2',
+  version = '0.0.3',
   license='MIT',
   description = 'Emergent Symbol Binding Network - Pytorch',
   author = 'Phil Wang',