diff --git a/src/TensorFlowNET.Core/Checkpoint/c_api.checkpoint.cs b/src/TensorFlowNET.Core/Checkpoint/c_api.checkpoint.cs
index e4563eb3..f956e333 100644
--- a/src/TensorFlowNET.Core/Checkpoint/c_api.checkpoint.cs
+++ b/src/TensorFlowNET.Core/Checkpoint/c_api.checkpoint.cs
@@ -8,7 +8,7 @@ namespace Tensorflow
         [DllImport(TensorFlowLibName)]
         internal static extern SafeCheckpointReaderHandle TF_NewCheckpointReader(string filename, SafeStatusHandle status);
         [DllImport(TensorFlowLibName)]
-        internal static extern void TF_DeleteCheckpointReader(SafeCheckpointReaderHandle reader);
+        internal static extern void TF_DeleteCheckpointReader(IntPtr reader);
         [DllImport(TensorFlowLibName)]
         internal static extern int TF_CheckpointReaderHasTensor(SafeCheckpointReaderHandle reader, string name);
         [DllImport(TensorFlowLibName)]