diff --git a/SAS/TMSS/test/tmss_test_data_django_models.py b/SAS/TMSS/test/tmss_test_data_django_models.py
index 15ba9e74bc44da03d6742c1cac15c21ebb6575bb..74827849fa6b08cce78318aff1cdf37d6d4d4781 100644
--- a/SAS/TMSS/test/tmss_test_data_django_models.py
+++ b/SAS/TMSS/test/tmss_test_data_django_models.py
@@ -287,7 +287,7 @@ def Dataproduct_test_data(producer: models.SubtaskOutput=None,
         producer = models.SubtaskOutput.objects.create(**SubtaskOutput_test_data())
 
     if dataformat is None:
-        dataformat = models.Dataformat.objects.get(value=models.Dataformat.Choices.MEASUREMENTSET)
+        dataformat = models.Dataformat.objects.get(value="MeasurementSet")
 
     return {"filename": filename,
             "directory": directory,