Fix test to expect correct format