diff --git a/tensorflow/lite/tools/optimize/calibration/node_info_delegate.cc b/tensorflow/lite/tools/optimize/calibration/node_info_delegate.cc
index 2b9197498b0..84031761b30 100644
--- a/tensorflow/lite/tools/optimize/calibration/node_info_delegate.cc
+++ b/tensorflow/lite/tools/optimize/calibration/node_info_delegate.cc
@@ -33,11 +33,14 @@ TfLiteStatus NodeInfoDelegatePrepare(TfLiteContext* context,
 }  // namespace
 
 TfLiteDelegate CreateNodeInfoDelegate(NodeInfoDelegateParams* params) {
-  return {/*data_ */ params,
-          /* Prepare */ NodeInfoDelegatePrepare,
-          /* CopyFromBufferHandle*/ nullptr,
-          /* CopyToBufferHandle*/ nullptr,
-          /* FreeBufferHandle*/ nullptr};
+  auto delegate = TfLiteDelegateCreate();
+  delegate.data_ = params;
+  delegate.Prepare = NodeInfoDelegatePrepare;
+  delegate.CopyFromBufferHandle = nullptr;
+  delegate.CopyToBufferHandle = nullptr;
+  delegate.FreeBufferHandle = nullptr;
+  delegate.flags = kTfLiteDelegateFlagsAllowDynamicTensors;
+  return delegate;
 }
 
 TfLiteStatus NodeInfoDelegateObserver::OnDelegatePrepareCalled(