Correct header inclusion