Update test files for new output