diff --git a/src/class/AILearning.js b/src/class/AILearning.js
index ef33aa09ff..fa4a76fd0c 100644
--- a/src/class/AILearning.js
+++ b/src/class/AILearning.js
@@ -250,7 +250,15 @@ export default class AILearning {
         }
         this.destroy();
 
-        this.#labels = labels || classes.map(({ name }) => name);
+        if (!labels) {
+            const parsed =
+                classes.length === 1 && typeof classes[0] === 'string'
+                    ? JSON.parse(classes[0])
+                    : classes;
+            this.#labels = parsed.map(({ name }) => name);
+        } else {
+            this.#labels = labels;
+        }
         this.#type = type;
         this.#url = url;
         this.#oid = _id;