add missing header changes for previous commit