diff --git a/projects/ml/sklearn-clj/notebooks/index.clj b/projects/ml/sklearn-clj/notebooks/index.clj
index 7bc801f..eb4e2ec 100644
--- a/projects/ml/sklearn-clj/notebooks/index.clj
+++ b/projects/ml/sklearn-clj/notebooks/index.clj
@@ -78,13 +78,11 @@ model-object
   (tc/head (tc/shuffle iris) 10) )
 
 (def prediction
-  (->
-   (:metamorph/data
+  (:metamorph/data
     (mm/transform-pipe 
      simulated-new-data  
      pipe-fn 
-     trained-ctx))
-   ds-cat/reverse-map-categorical-xforms))
+     trained-ctx)))
 
 ;; We get a tech.ml.dataset with the prediction result back.
 ;; `sklearn-clj` auto-transform the prediction result back to a