diff --git a/reasoning_gym/code/codeio.py b/reasoning_gym/code/codeio.py
index 258d5e03..f71555f6 100644
--- a/reasoning_gym/code/codeio.py
+++ b/reasoning_gym/code/codeio.py
@@ -173,7 +173,7 @@ class CodeIODataset(ProceduralDataset):
                         # TODO: Consider a more sophisticated distance metric for numeric values?
                         abs1, abs2 = abs(float(value1)), abs(float(value2))
                         divisor = max(min(abs1, abs2), 10e-5)
-                        value_dist += (abs1 - abs2) / divisor
+                        value_dist = max(len(value1), len(value2)) * abs((abs1 - abs2) / divisor)
                     except ValueError:
                         # Fall back on string edit distance
                         pass
diff --git a/reasoning_gym/data/codeio.jsonl.gz b/reasoning_gym/data/codeio.jsonl.gz
index 70ba8d9d..e4d0891b 100644
Binary files a/reasoning_gym/data/codeio.jsonl.gz and b/reasoning_gym/data/codeio.jsonl.gz differ