fixing return type prob