add missing header

fix type