diff --git a/src/memory_tracker-inl.h b/src/memory_tracker-inl.h
index 1a28e2dd792747..9e6201442ab6b1 100644
--- a/src/memory_tracker-inl.h
+++ b/src/memory_tracker-inl.h
@@ -107,9 +107,9 @@ void MemoryTracker::TrackField(const char* edge_name,
   }
 }
 
-template <typename T>
+template <typename T, typename D>
 void MemoryTracker::TrackField(const char* edge_name,
-                               const std::unique_ptr<T>& value,
+                               const std::unique_ptr<T, D>& value,
                                const char* node_name) {
   if (value.get() == nullptr) {
     return;
diff --git a/src/memory_tracker.h b/src/memory_tracker.h
index 616976ab2afab1..4a66e9ce74ccc5 100644
--- a/src/memory_tracker.h
+++ b/src/memory_tracker.h
@@ -140,9 +140,9 @@ class MemoryTracker {
                                        const char* node_name = nullptr);
 
   // Shortcut to extract the underlying object out of the smart pointer
-  template <typename T>
+  template <typename T, typename D>
   inline void TrackField(const char* edge_name,
-                         const std::unique_ptr<T>& value,
+                         const std::unique_ptr<T, D>& value,
                          const char* node_name = nullptr);
 
   template <typename T>