partial fix
fixed type error