Minor fix in tests.
Started messing with Diff.