Fix output checks