diff --git a/hourglass_tensorflow/handlers/dataset.py b/hourglass_tensorflow/handlers/dataset.py
index a1f50c1..2a0a54f 100644
--- a/hourglass_tensorflow/handlers/dataset.py
+++ b/hourglass_tensorflow/handlers/dataset.py
@@ -71,7 +71,7 @@ def heatmap(self) -> HTFDatasetHeatmap:
 
     def select_engine(self, data: Any) -> HTFEngine:
         try:
-            self.engine = self._engines[type(data)](metadata=self._metadata)
+            return self._engines[type(data)](metadata=self._metadata)
         except KeyError:
             raise KeyError(f"No engine available for type {type(data)}")