diff --git a/packages/data-designer-config/src/data_designer/config/dataset_metadata.py b/packages/data-designer-config/src/data_designer/config/dataset_metadata.py
index 8b4f4a329..fe7c19aaa 100644
--- a/packages/data-designer-config/src/data_designer/config/dataset_metadata.py
+++ b/packages/data-designer-config/src/data_designer/config/dataset_metadata.py
@@ -1,6 +1,8 @@
 # SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 from pydantic import BaseModel