Fix header class