Simplify provided dataset in tests