diff --git a/tests/dummy_datastore.py b/tests/dummy_datastore.py
index 5525f36..0c76bca 100644
--- a/tests/dummy_datastore.py
+++ b/tests/dummy_datastore.py
@@ -332,6 +332,8 @@ def get_dataarray(
             The category of the dataset (state/forcing/static).
         split : str
             The time split to filter the dataset (train/val/test).
+        standardize: bool
+            If the dataarray should be returned standardized
 
         Returns
         -------