Fix header file include