Update tests to reflect header schema changes.