fix missing type bug