diff --git a/morph_net/tools/structure_exporter.py b/morph_net/tools/structure_exporter.py
index ca17479..17f17e7 100644
--- a/morph_net/tools/structure_exporter.py
+++ b/morph_net/tools/structure_exporter.py
@@ -70,8 +70,7 @@ def __init__(self,
         tf.logging.warning('No regularizer found for: %s', op.name)
         continue
 
-      self._tensors[rename_fn(op.name)] = tf.cast(op_regularizer.alive_vector,
-                                                  tf.int32)
+      self._tensors[rename_fn(op.name)] = op_regularizer.alive_vector
 
   @property
   def tensors(self):