diff --git a/CHANGELOG.md b/CHANGELOG.md
index 4fa9831..909c046 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -16,3 +16,4 @@
 
 ## BUGFIXES
 
+* Updated `file_prediction.yaml` in `api` (PR #3).
diff --git a/src/api/file_prediction.yaml b/src/api/file_prediction.yaml
index 26068ab..7a81391 100644
--- a/src/api/file_prediction.yaml
+++ b/src/api/file_prediction.yaml
@@ -1,15 +1,14 @@
-#TODO: Change to the required and/or optional fields of the anndata
 type: file
-example: "resources_test/task_template/cxg_mouse_pancreas_atlas/prediction.h5ad"
+example: "resources_test/task_spatial_trajectory_inference/cxg_mouse_pancreas_atlas/prediction.h5ad"
 label: "Predicted data"
 summary: A predicted dataset as output by a method.
 info:
   format:
     type: h5ad
     obs:
-      - type: string
-        name: label_pred
-        description: Predicted labels for the test cells.
+      - type: double
+        name: pseudotime_inferred
+        description: Predicted pseudotime for each cell.
         required: true
     uns:
       - type: string