diff --git a/combine.py b/combine.py
index 10b142ed52969dd3f5e072b65ab7ec0a39e7cb9c..3f588c8da5aaa0badef9f39b807c47185e37be2e 100644
--- a/combine.py
+++ b/combine.py
@@ -8,12 +8,15 @@ def combine(BF_2D_path:str, TRITC_3D_path:str, out_zarr_path:str):
     bd2d = nd2.ND2File(BF_2D_path).to_dask()
     print('Opened BF:', bd2d)
     fd3d = nd2.ND2File(TRITC_3D_path).to_dask()
+    z_axis = nd2.ND2File(TRITC_3D_path).sizes.index("Z")
     print('Opened TRITC:', fd3d)
-    fd2d = fd3d.max(axis=1)
-    bd2d = da.stack([bd2d, fd2d], axis=1)
+    fd2d = fd3d.max(axis=z_axis)
+    c_axis = fd2d.ndim - 2
+    bd2d = da.stack([bd2d, fd2d], axis=c_axis)
     print('Resulting stack:', bd2d)
     to_zarr(
-        bd2d, 
+        bd2d,
+        channel_axis=c_axis, 
         path=out_zarr_path, 
         steps=4, 
         name=['BF','TRITC'],