diff --git a/src/TensorFlowNET.Core/Training/Optimizer.cs b/src/TensorFlowNET.Core/Training/Optimizer.cs
index 524a0e34..04ec949c 100644
--- a/src/TensorFlowNET.Core/Training/Optimizer.cs
+++ b/src/TensorFlowNET.Core/Training/Optimizer.cs
@@ -205,7 +205,9 @@ namespace Tensorflow
                         //}
                         //else
                         {
-                            apply_updates = state_ops.assign_add(global_step, tf.constant(1), name: name);
+                            apply_updates = state_ops.assign_add(global_step,
+                                ops.convert_to_tensor(1, dtype: global_step.dtype),
+                                name: name);
                         }
                     });
                 }
diff --git a/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj b/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
index 76965c1e..64ca76f6 100644
--- a/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
+++ b/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
@@ -2,7 +2,7 @@
   <PropertyGroup>
     <RootNamespace>Tensorflow.Hub</RootNamespace>
     <TargetFramework>netstandard2.0</TargetFramework>
-    <Version>0.0.6</Version>
+    <Version>0.1.0</Version>
     <Authors>Kerry Jiang, Haiping Chen</Authors>
     <Company>SciSharp STACK</Company>
     <Copyright>Apache 2.0</Copyright>