Fix tests for headings