diff --git a/sonnet/src/nets/dnc/read.py b/sonnet/src/nets/dnc/read.py
index 7c1bde6..a05821c 100644
--- a/sonnet/src/nets/dnc/read.py
+++ b/sonnet/src/nets/dnc/read.py
@@ -36,7 +36,7 @@ def read(memory,
   """
   with tf.name_scope("read_memory"):
     if squash_before_access:
-      squash_op(weights)
+      weights = squash_op(weights)
     read_word = tf.matmul(weights, memory)
     if squash_after_access:
       read_word = squash_op(read_word)