diff --git a/models/ops.py b/models/ops.py
index 88af80f..1fde044 100644
--- a/models/ops.py
+++ b/models/ops.py
@@ -21,6 +21,7 @@ def _next_inputs_fn(self, time, outputs, state, sample_ids):
         return (finished, next_inputs, state)
 
     def __init__(self, batch_size, out_size):
+        super().__init__(self._initialize_fn, self._sample_fn, self._next_inputs_fn)
         self._batch_size = batch_size
         self._out_size = out_size