diff --git a/src/TensorFlowNET.Console/Tensorflow.Console.csproj b/src/TensorFlowNET.Console/Tensorflow.Console.csproj
index 8efbf1bb..2ed2f41b 100644
--- a/src/TensorFlowNET.Console/Tensorflow.Console.csproj
+++ b/src/TensorFlowNET.Console/Tensorflow.Console.csproj
@@ -19,7 +19,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.5.0" />
+    <PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" />
   </ItemGroup>
 
   <ItemGroup>
diff --git a/src/TensorFlowNET.Core/Operations/Distributions/distribution.py.cs b/src/TensorFlowNET.Core/Operations/Distributions/distribution.py.cs
index 988c5326..4375788d 100644
--- a/src/TensorFlowNET.Core/Operations/Distributions/distribution.py.cs
+++ b/src/TensorFlowNET.Core/Operations/Distributions/distribution.py.cs
@@ -60,25 +60,7 @@ namespace Tensorflow
         {
             return tf_with(ops.name_scope(name, "moments", new { value }), scope =>
             {
-                try
-                {
-                    return _log_prob(value);
-                }
-#pragma warning disable CS0168 // Variable is declared but never used
-                catch (Exception e1)
-#pragma warning restore CS0168 // Variable is declared but never used
-                {
-                    try
-                    {
-                        return math_ops.log(_prob(value));
-#pragma warning disable CS0168 // Variable is declared but never used
-                    }
-                    catch (Exception e2)
-#pragma warning restore CS0168 // Variable is declared but never used
-                    {
-                        throw new NotImplementedException();
-                    }
-                }
+                return math_ops.log(value);
             });
         }
 
diff --git a/src/TensorFlowNET.Core/Operations/array_ops.cs b/src/TensorFlowNET.Core/Operations/array_ops.cs
index fc83cf4e..b0ef1f2d 100644
--- a/src/TensorFlowNET.Core/Operations/array_ops.cs
+++ b/src/TensorFlowNET.Core/Operations/array_ops.cs
@@ -656,7 +656,7 @@ namespace Tensorflow
         /// <param name="name"></param>
         /// <returns></returns>
         public static Tensor stop_gradient(Tensor input, string name = null)
-            => gen_array_ops.stop_gradient(input, name);
+            => tf.Context.ExecuteOp("StopGradient", name, new ExecuteOpArgs(input));
 
         /// <summary>
         /// Extracts a strided slice of a tensor (generalized python array indexing).