[Algorithm] Small fix