diff --git a/src/everest/export.py b/src/everest/export.py
index ac49b082d92..9cf27b2e9dd 100644
--- a/src/everest/export.py
+++ b/src/everest/export.py
@@ -157,6 +157,7 @@ def get_internalized_keys(
             assert len(experiments) == 1
             experiment = experiments[0]
 
+            ensemble = experiment.get_ensemble_by_name(f"batch_{batch_id}")
             ensemble = experiment.get_ensemble_by_name(f"batch_{batch_id}")
             if not internal_keys:
                 internal_keys = set(