Fix header inclusion in ui files.