Fixed rouge return type