TSAGE: Added missing header change from previous commit