Fix broken measure_diff_test