diff --git a/src/TensorFlowNET.Core/Gradients/TapeTensor.cs b/src/TensorFlowNET.Core/Gradients/TapeTensor.cs
index 92c4e39f..be030321 100644
--- a/src/TensorFlowNET.Core/Gradients/TapeTensor.cs
+++ b/src/TensorFlowNET.Core/Gradients/TapeTensor.cs
@@ -22,5 +22,8 @@ namespace Tensorflow.Gradients
 
         public Tensor OnesLike()
             => tf.ones(shape: shape, dtype: dtype);
+
+        public override string ToString()
+            => $"{id}, {shape}, {dtype.as_numpy_name()}";
     }
 }
diff --git a/src/TensorFlowNET.Keras/Layers/BatchNormalization.cs b/src/TensorFlowNET.Keras/Layers/BatchNormalization.cs
index a160e496..18bd5c55 100644
--- a/src/TensorFlowNET.Keras/Layers/BatchNormalization.cs
+++ b/src/TensorFlowNET.Keras/Layers/BatchNormalization.cs
@@ -196,7 +196,7 @@ namespace Tensorflow.Keras.Layers
                     _assign_moving_average(moving_variance, variance, momentum_tensor);
 
                 if (use_fused_avg_updates)
-                    _assign_new_value(moving_variance, mean);
+                    _assign_new_value(moving_variance, variance);
                 else
                     _assign_moving_average(moving_variance, variance, momentum_tensor);