diff --git a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs
index e8e87840..b3729abc 100644
--- a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs
+++ b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs
@@ -65,7 +65,7 @@ namespace TensorFlowNET.UnitTest.Dataset
         {
             var X = new[] { 2013, 2014, 2015, 2016, 2017 };
 
-            var dataset = tf.data.Dataset.from_tensor(X);
+            var dataset = tf.data.Dataset.from_tensors(X);
             int n = 0;
             foreach (var x in dataset)
             {