fixing return type