Header include fix