fix types issue