diff --git a/src/core/modeling/modeling_batch.py b/src/core/modeling/modeling_batch.py
index c61737d..497a3e2 100644
--- a/src/core/modeling/modeling_batch.py
+++ b/src/core/modeling/modeling_batch.py
@@ -795,8 +795,9 @@ class WaterQualityModelingBatch:
                     _phys.wavelengths, dtype=np.float64
                 )
                 _feat_map = {}
-                for name, ia, ib in _phys._feat_cols_:
+                for name, ftype, ia, ib in _phys._feat_cols_:
                     _feat_map[name] = {
+                        'ftype': str(ftype),
                         'wl_a': float(_phys.wavelengths[ia]),
                         'wl_b': float(_phys.wavelengths[ib]),
                         'idx_a': int(ia),