diff --git a/draco/core/containers.py b/draco/core/containers.py
index d004463fc..d3fcd3307 100644
--- a/draco/core/containers.py
+++ b/draco/core/containers.py
@@ -219,7 +219,7 @@ def __init__(self, *args, **kwargs):
         # Iterate over datasets and initialise any that specify it
         if not skip_datasets:
             for name, spec in self.dataset_spec.items():
-                if "initialise" in spec and spec["initialise"]:
+                if spec.get("initialise"):
                     self.add_dataset(name)
 
         # Copy over datasets that have compatible axes
diff --git a/draco/synthesis/noise.py b/draco/synthesis/noise.py
index 4ae9e1365..345431c91 100644
--- a/draco/synthesis/noise.py
+++ b/draco/synthesis/noise.py
@@ -211,7 +211,7 @@ def process(self, data):
 
         if self.add_noise:
             noise = random.complex_normal(
-                (nfreq, nprod, ntime),
+                size=(nfreq, nprod, ntime),
                 scale=std[np.newaxis, :, np.newaxis],
                 rng=self.rng,
             )