Fix test to fit app output