fix incorrect return type :\