Update unit tests to reflect header schema changes.