diff --git a/torchcast/__init__.py b/torchcast/__init__.py
index 2b8877c..93b60a1 100644
--- a/torchcast/__init__.py
+++ b/torchcast/__init__.py
@@ -1 +1 @@
-__version__ = '0.5.0'
+__version__ = '0.5.1'
diff --git a/torchcast/state_space/predictions.py b/torchcast/state_space/predictions.py
index 7dd6239..0e49e82 100644
--- a/torchcast/state_space/predictions.py
+++ b/torchcast/state_space/predictions.py
@@ -463,7 +463,11 @@ def _tensor_to_df(tens, measures):
             raise ValueError("Expected `type` to be 'predictions' or 'components'.")
 
         out = pd.concat(out).reset_index(drop=True)
-        _out_cols = [group_colname, time_colname, 'measure', 'mean', 'lower', 'upper']
+        _out_cols = [group_colname, time_colname, 'measure', 'mean']
+        if conf is None:
+            _out_cols.append('std')
+        else:
+            _out_cols.extend(['lower', 'upper'])
         if type.startswith('comp'):
             _out_cols = _out_cols[0:3] + ['process', 'state_element'] + _out_cols[3:]
         if 'actual' in out.columns: