Fixing return type.