diff --git a/src/TensorFlowNET.Core/Buffers/c_api.buffer.cs b/src/TensorFlowNET.Core/Buffers/c_api.buffer.cs
index a7dc0ff2..2e242230 100644
--- a/src/TensorFlowNET.Core/Buffers/c_api.buffer.cs
+++ b/src/TensorFlowNET.Core/Buffers/c_api.buffer.cs
@@ -32,7 +32,7 @@ namespace Tensorflow
         public static extern SafeBufferHandle TF_NewBuffer();
 
         [DllImport(TensorFlowLibName)]
-        public static extern IntPtr TF_GetBuffer(TF_Buffer buffer);
+        public static extern TF_Buffer TF_GetBuffer(SafeBufferHandle buffer);
 
         /// <summary>
         /// Makes a copy of the input and sets an appropriate deallocator.  Useful for