Fixed broken test with new format