Minor update to expected output