diff --git a/containers/popv/Dockerfile b/containers/popv/Dockerfile
index 1467a68..d3415e3 100644
--- a/containers/popv/Dockerfile
+++ b/containers/popv/Dockerfile
@@ -1,6 +1,6 @@
 FROM tensorflow/tensorflow:2.16.1
 
-RUN apt-get -y update && apt-get -y install python3-dev python3-pip g++ git wget && apt-get clean
+RUN apt-get -y update && apt-get -y install git && apt-get clean
 
 COPY context/download-ontology.sh .
 RUN ./download-ontology.sh
diff --git a/containers/popv/context/main.py b/containers/popv/context/main.py
index b6e357a..1fa8549 100644
--- a/containers/popv/context/main.py
+++ b/containers/popv/context/main.py
@@ -126,14 +126,12 @@ def do_run(
         data = self.prepare_query(data, organ, metadata["model"], options)
         popv.annotation.annotate_data(
             data,
-            # TODO: onclass has been removed due to error in fast mode
-            # seen_result_key is not added to the result in fast mode but still expected during compute_consensus
             methods=[
                 "knn_on_scvi",
                 "scanvi",
                 "svm",
                 "rf",
-                # "onclass",
+                "onclass",
                 "celltypist",
             ],
         )