diff --git a/tsfm_public/toolkit/util.py b/tsfm_public/toolkit/util.py
index b305c80b..29fb7cf0 100644
--- a/tsfm_public/toolkit/util.py
+++ b/tsfm_public/toolkit/util.py
@@ -648,6 +648,7 @@ class values.
                         )
                     has_class_labels_tag = True
                     metadata_started = True
+                    classification_dataset = False
                 # Check if this line contains the start of data
                 elif line.startswith("@data"):
                     if line != "@data":