diff --git a/test/problems/divsort/generators/generators.yaml b/test/problems/divsort/generators/generators.yaml
index 2883b035..94d8d01d 100644
--- a/test/problems/divsort/generators/generators.yaml
+++ b/test/problems/divsort/generators/generators.yaml
@@ -31,8 +31,9 @@ data:
           'nested_1':
             testdata.yaml:
               input_validator_flags: "--small"
-            include:
-              - 'small_floats'
+            data:
+              'small_floats':
+                in: 10 3.5 ab cd
           'nested_2':
             testdata.yaml:
               input_validator_flags:
@@ -61,7 +62,8 @@ data:
           input_validator_flags: --integer
         data:
           ints_expected: { in: 10.0 2.5 ab cd }
-          small_floats: { in: 10.0 3.5 ab cd }
+        include:
+          - small_floats
       sorted:
         testdata.yaml:
           input_validator_flags: --sorted