Fix a header inclusion error