diff --git a/pyproject.toml b/pyproject.toml
index 01501c6..1179169 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quera-ahs-utils"
-version = "0.4.3"
+version = "0.4.4"
 authors = [
   { name="Phillip Weinberg", email="pweinberg@quera.com" },
   { name="John Long", email="jlong@quera.com" }
@@ -26,7 +26,7 @@ dependencies = [
   "numpy",
   "scipy",
   "matplotlib",
-  "pydantic",
+  "pydantic>=2.0.0",
 ]
 
 [project.urls]
diff --git a/src/quera_ahs_utils/quera_ir/capabilities.py b/src/quera_ahs_utils/quera_ir/capabilities.py
index e86a55f..1e1ca26 100644
--- a/src/quera_ahs_utils/quera_ir/capabilities.py
+++ b/src/quera_ahs_utils/quera_ir/capabilities.py
@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 __all__ = ["QuEraCapabilities"]
 
diff --git a/src/quera_ahs_utils/quera_ir/task_results.py b/src/quera_ahs_utils/quera_ir/task_results.py
index be51f4c..3482a4b 100644
--- a/src/quera_ahs_utils/quera_ir/task_results.py
+++ b/src/quera_ahs_utils/quera_ir/task_results.py
@@ -1,5 +1,5 @@
 from enum import Enum
-from pydantic import BaseModel, conlist, conint
+from pydantic.v1 import BaseModel, conlist, conint
 from typing import Callable, Optional, List, Tuple
 import numpy as np
 
diff --git a/src/quera_ahs_utils/quera_ir/task_specification.py b/src/quera_ahs_utils/quera_ir/task_specification.py
index e9a1bfe..492008d 100644
--- a/src/quera_ahs_utils/quera_ir/task_specification.py
+++ b/src/quera_ahs_utils/quera_ir/task_specification.py
@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 from typing import Optional, List, Tuple, Union
 from decimal import Decimal
 
@@ -155,6 +155,7 @@ class QuEraTaskSpecification(BaseModel):
     nshots: int
     lattice: Lattice
     effective_hamiltonian: EffectiveHamiltonian
+    metadata: Optional[str] = None
     
     def __hash__(self):
         return hash((QuEraTaskSpecification, self.nshots, self.lattice, self.effective_hamiltonian))
@@ -163,6 +164,7 @@ def discretize(self, task_capabilities: QuEraCapabilities):
         return QuEraTaskSpecification(
             nshots = self.nshots,
             lattice = self.lattice.discretize(task_capabilities),
-            effective_hamiltonian = self.effective_hamiltonian.discretize(task_capabilities)
+            effective_hamiltonian = self.effective_hamiltonian.discretize(task_capabilities),
+            metadata=self.metadata,
         )