diff --git a/htc/scikit-learn:1.3/Dockerfile b/htc/scikit-learn:1.3/Dockerfile
index 7bdcfca..21ad939 100644
--- a/htc/scikit-learn:1.3/Dockerfile
+++ b/htc/scikit-learn:1.3/Dockerfile
@@ -3,7 +3,7 @@ FROM hub.opensciencegrid.org/htc/ubuntu:22.04
 LABEL opensciencegrid.name="scikit-learn:1.3.2"
 LABEL opensciencegrid.description="scikit-learn, configured for execution on OSG"
 LABEL opensciencegrid.url="https://scikit-learn.org/"
-LABEL opensciencegrid.category="Base"
+LABEL opensciencegrid.category="AI"
 LABEL opensciencegrid.definition_url="https://github.com/osg-htc/htc-images"
 
 COPY 90-scikit-learn.sh /.singularity.d/env/90-scikit-learn.sh
diff --git a/htc/tensorflow:2.15/Dockerfile b/htc/tensorflow:2.15/Dockerfile
index f33a88d..1a271b1 100644
--- a/htc/tensorflow:2.15/Dockerfile
+++ b/htc/tensorflow:2.15/Dockerfile
@@ -3,7 +3,7 @@ FROM tensorflow/tensorflow:2.15.0-gpu
 LABEL opensciencegrid.name="Tensorflow 2.15"
 LABEL opensciencegrid.description="Tensorflow image from the Tensorflow project, with OSG additions"
 LABEL opensciencegrid.url="https://www.tensorflow.org/"
-LABEL opensciencegrid.category="Base"
+LABEL opensciencegrid.category="AI"
 LABEL opensciencegrid.definition_url="https://github.com/osg-htc/htc-images"
 
 RUN export DEBIAN_FRONTEND=noninteractive DEBCONF_NONINTERACTIVE_SEEN=true && \