diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
index 003a7fd9..34edcb4f 100644
--- a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
+++ b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
@@ -523,7 +523,7 @@ namespace Tensorflow
             if (TF_TensorData(handle).ToPointer() == arraySlice.Address)
             {
                 AllocationType = AllocationType.FromPointer;
-                AllocationReferenceHolder = nd.Unsafe.Storage;
+                AllocationReferenceHolder = arraySlice;
             } else
                 AllocationType = AllocationType.Tensorflow;