Fix test that fails because of pretty printing