[Driver] Add missed include header.