Added missing header import