diff --git a/docs/source/conf.py b/docs/source/conf.py
index 4e7bfb1b..68b6f5ae 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -24,9 +24,9 @@ copyright = '2019, Haiping Chen'
 author = 'Haiping Chen'
 
 # The short X.Y version
-version = '0.0.1'
+version = '0.6.0'
 # The full version, including alpha/beta/rc tags
-release = '0.0.1'
+release = '0.6.0'
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/src/TensorFlowNET.Core/Sessions/BaseSession.cs b/src/TensorFlowNET.Core/Sessions/BaseSession.cs
index 530a0107..07c4024a 100644
--- a/src/TensorFlowNET.Core/Sessions/BaseSession.cs
+++ b/src/TensorFlowNET.Core/Sessions/BaseSession.cs
@@ -245,6 +245,8 @@ namespace Tensorflow
                     throw new NotImplementedException("can't fetch output");
             }
 
+            tensor.Dispose();
+
             return nd;
         }
 
diff --git a/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj b/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj
index 30ce8513..a4188e15 100644
--- a/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj
+++ b/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj
@@ -4,7 +4,7 @@
     <TargetFramework>netstandard2.0</TargetFramework>
     <AssemblyName>TensorFlow.NET</AssemblyName>
     <RootNamespace>Tensorflow</RootNamespace>
-    <Version>0.6.0</Version>
+    <Version>0.5.2</Version>
     <Authors>Haiping Chen</Authors>
     <Company>SciSharp STACK</Company>
     <GeneratePackageOnBuild>false</GeneratePackageOnBuild>
@@ -14,16 +14,14 @@
     <PackageProjectUrl>https://github.com/SciSharp</PackageProjectUrl>
     <PackageIconUrl>https://avatars3.githubusercontent.com/u/44989469?s=200&amp;v=4</PackageIconUrl>
     <PackageTags>TensorFlow, NumSharp, SciSharp, MachineLearning, TensorFlow.NET, C#</PackageTags>
-    <Description>Google's TensorFlow binding in .NET Standard.
+    <Description>Google's TensorFlow full binding in .NET Standard.
 Docs: https://tensorflownet.readthedocs.io</Description>
-    <AssemblyVersion>0.6.0.0</AssemblyVersion>
+    <AssemblyVersion>0.5.2.0</AssemblyVersion>
     <PackageReleaseNotes>Changes since v0.5:
-Added K-means Clustering.
 Added Nearest Neighbor.
-Added a lot of APIs to build neural networks model.
-Bug fix.</PackageReleaseNotes>
+Bug memory leak issue when allocating Tensor.</PackageReleaseNotes>
     <LangVersion>7.2</LangVersion>
-    <FileVersion>0.6.0.0</FileVersion>
+    <FileVersion>0.5.2.0</FileVersion>
   </PropertyGroup>
 
   <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|AnyCPU'">
diff --git a/test/TensorFlowNET.Examples/TensorFlowNET.Examples.csproj b/test/TensorFlowNET.Examples/TensorFlowNET.Examples.csproj
index 59fc1fd0..912dc91c 100644
--- a/test/TensorFlowNET.Examples/TensorFlowNET.Examples.csproj
+++ b/test/TensorFlowNET.Examples/TensorFlowNET.Examples.csproj
@@ -3,6 +3,7 @@
   <PropertyGroup>
     <OutputType>Exe</OutputType>
     <TargetFramework>netcoreapp2.2</TargetFramework>
+    <GeneratePackageOnBuild>false</GeneratePackageOnBuild>
   </PropertyGroup>
 
   <ItemGroup>