diff --git a/src/TensorFlowNET.Core/Data/MnistModelLoader.cs b/src/TensorFlowNET.Core/Data/MnistModelLoader.cs
index 0aaeb32a..50acb4ba 100644
--- a/src/TensorFlowNET.Core/Data/MnistModelLoader.cs
+++ b/src/TensorFlowNET.Core/Data/MnistModelLoader.cs
@@ -123,7 +123,7 @@ namespace Tensorflow
 
                 bytestream.Read(buf, 0, buf.Length);
 
-                var data = np.frombuffer(buf, (num_images, rows * cols), np.@byte);
+                var data = np.frombuffer(buf, (num_images, rows * cols), np.uint8);
                 return data;
             }
         }
@@ -146,7 +146,7 @@ namespace Tensorflow
 
                 bytestream.Read(buf, 0, buf.Length);
 
-                var labels = np.frombuffer(buf, new Shape(num_items), np.@byte);
+               var labels = np.frombuffer(buf, new Shape(num_items), np.uint8);
 
                 if (one_hot)
                     return DenseToOneHot(labels, num_classes);
diff --git a/src/TensorFlowNET.Core/Operations/array_ops.cs b/src/TensorFlowNET.Core/Operations/array_ops.cs
index 287bc37a..26bbd042 100644
--- a/src/TensorFlowNET.Core/Operations/array_ops.cs
+++ b/src/TensorFlowNET.Core/Operations/array_ops.cs
@@ -91,6 +91,9 @@ namespace Tensorflow
                             zeros = constant(0f);
                             break;
                         case TF_DataType.TF_INT8:
+                            zeros = constant((sbyte)0);
+                            break;
+                        case TF_DataType.TF_UINT8:
                             zeros = constant((byte)0);
                             break;
                         default: