Fix header inclusion