Fix test for output