Fix a missing header