diff --git a/src/TensorFlowNET.Keras/Tensorflow.Keras.csproj b/src/TensorFlowNET.Keras/Tensorflow.Keras.csproj
index 54d45461..40cdf961 100644
--- a/src/TensorFlowNET.Keras/Tensorflow.Keras.csproj
+++ b/src/TensorFlowNET.Keras/Tensorflow.Keras.csproj
@@ -7,7 +7,7 @@
     <Nullable>enable</Nullable>
     <RootNamespace>Tensorflow.Keras</RootNamespace>
     <Platforms>AnyCPU;x64</Platforms>
-    <Version>0.6.3</Version>
+    <Version>0.6.4</Version>
     <Authors>Haiping Chen</Authors>
     <Product>Keras for .NET</Product>
     <Copyright>Apache 2.0, Haiping Chen 2021</Copyright>
@@ -24,9 +24,9 @@
 * Implemented backward_function.
 * Support model.load_weights.
 * Add Subtract layer
-* Support YOLOv3 model.
 * Text preprocessing
-* Preprocessing.timeseries_dataset_from_array</PackageReleaseNotes>
+* Preprocessing.timeseries_dataset_from_array
+* Fixed memory leak for YOLOv3 model.</PackageReleaseNotes>
     <Description>Keras for .NET
 
 Keras is an API designed for human beings, not machines. Keras follows best practices for reducing cognitive load: it offers consistent &amp; simple APIs, it minimizes the number of user actions required for common use cases, and it provides clear &amp; actionable error messages.</Description>
@@ -37,8 +37,8 @@ Keras is an API designed for human beings, not machines. Keras follows best prac
     <RepositoryType>Git</RepositoryType>
     <SignAssembly>true</SignAssembly>
     <AssemblyOriginatorKeyFile>Open.snk</AssemblyOriginatorKeyFile>
-    <AssemblyVersion>0.6.3.0</AssemblyVersion>
-    <FileVersion>0.6.3.0</FileVersion>
+    <AssemblyVersion>0.6.4.0</AssemblyVersion>
+    <FileVersion>0.6.4.0</FileVersion>
     <PackageLicenseFile>LICENSE</PackageLicenseFile>
   </PropertyGroup>
 
diff --git a/test/TensorFlowNET.Graph.UnitTest/MultithreadingTests.cs b/test/TensorFlowNET.Graph.UnitTest/MultithreadingTests.cs
index fad6196b..1577f6f1 100644
--- a/test/TensorFlowNET.Graph.UnitTest/MultithreadingTests.cs
+++ b/test/TensorFlowNET.Graph.UnitTest/MultithreadingTests.cs
@@ -109,9 +109,6 @@ namespace TensorFlowNET.UnitTest
         [TestMethod]
         public void TensorCreation()
         {
-            //lock (Locks.ProcessWide)
-            //    tf.Session(); //create one to increase next id to 1.
-
             MultiThreadedUnitTestExecuter.Run(8, Core);
 
             //the core method
@@ -131,9 +128,6 @@ namespace TensorFlowNET.UnitTest
         [TestMethod]
         public void TensorCreation_Array()
         {
-            //lock (Locks.ProcessWide)
-            //    tf.Session(); //create one to increase next id to 1.
-
             MultiThreadedUnitTestExecuter.Run(8, Core);
 
             //the core method
@@ -150,33 +144,6 @@ namespace TensorFlowNET.UnitTest
             }
         }
 
-        [TestMethod]
-        public void TensorCreation_Undressed()
-        {
-            //lock (Locks.ProcessWide)
-            //    tf.Session(); //create one to increase next id to 1.
-
-            MultiThreadedUnitTestExecuter.Run(8, Core);
-
-            //the core method
-            unsafe void Core(int tid)
-            {
-                using (var sess = tf.Session())
-                {
-                    for (int i = 0; i < 100; i++)
-                    {
-                        var v = (int*)Marshal.AllocHGlobal(sizeof(int));
-                        c_api.DeallocatorArgs _deallocatorArgs = new c_api.DeallocatorArgs();
-                        var handle = c_api.TF_NewTensor(typeof(int).as_tf_dtype(), dims: new long[0], num_dims: 0,
-                            data: (IntPtr)v, len: (UIntPtr)sizeof(int),
-                            deallocator: (IntPtr data, IntPtr size, ref c_api.DeallocatorArgs args) => Marshal.FreeHGlobal(data),
-                            ref _deallocatorArgs);
-                        c_api.TF_DeleteTensor(handle);
-                    }
-                }
-            }
-        }
-
         [TestMethod]
         public void SessionRun()
         {
diff --git a/test/TensorFlowNET.Graph.UnitTest/TensorFlowNET.Graph.UnitTest.csproj b/test/TensorFlowNET.Graph.UnitTest/TensorFlowNET.Graph.UnitTest.csproj
index eb5f9099..6112fc3b 100644
--- a/test/TensorFlowNET.Graph.UnitTest/TensorFlowNET.Graph.UnitTest.csproj
+++ b/test/TensorFlowNET.Graph.UnitTest/TensorFlowNET.Graph.UnitTest.csproj
@@ -15,7 +15,7 @@
 
   <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
     <DefineConstants>DEBUG;TRACE</DefineConstants>
-    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+    <AllowUnsafeBlocks>false</AllowUnsafeBlocks>
   </PropertyGroup>
 
   <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|AnyCPU'">
diff --git a/test/TensorFlowNET.Native.UnitTest/Tensorflow.Native.UnitTest.csproj b/test/TensorFlowNET.Native.UnitTest/Tensorflow.Native.UnitTest.csproj
index 78189777..ae809b6c 100644
--- a/test/TensorFlowNET.Native.UnitTest/Tensorflow.Native.UnitTest.csproj
+++ b/test/TensorFlowNET.Native.UnitTest/Tensorflow.Native.UnitTest.csproj
@@ -24,6 +24,10 @@
     <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
   </PropertyGroup>
 
+  <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+  </PropertyGroup>
+
   <ItemGroup>
     <None Remove="Lite\testdata\add.bin" />
     <None Remove="Lite\testdata\add_quantized.bin" />