diff --git a/matsciml/datasets/materials_project/dataset.py b/matsciml/datasets/materials_project/dataset.py
index 763a4308..9913ae7a 100644
--- a/matsciml/datasets/materials_project/dataset.py
+++ b/matsciml/datasets/materials_project/dataset.py
@@ -146,6 +146,7 @@ def _parse_structure(
             "lattice_params": lattice_params,
         }
         return_dict["lattice_features"] = lattice_features
+        return_dict["cell"] = structure.lattice.matrix
 
     def _parse_symmetry(
         self,