diff --git a/diagonals/ohc.py b/diagonals/ohc.py
index fb837548300b7a6c461d9791f70b1a16958c03ad..d08f26dcff67870493cc595f59cb943da3e549b5 100644
--- a/diagonals/ohc.py
+++ b/diagonals/ohc.py
@@ -163,7 +163,7 @@ def _compute_ohc_cpu(layers, thetao, weights, area):
     """
     ohc = []
     for layer in range(len(layers)):
-        ohc_layer = da.sum(
+        ohc_layer = da.nansum(
             _multiply_array(thetao, weights[layer]),
             axis=1
         )
@@ -171,7 +171,7 @@ def _compute_ohc_cpu(layers, thetao, weights, area):
         ohc1d_total = []
         for i, basin in enumerate(area):
             ohc_basin = _multiply_array(ohc_layer, area[basin])
-            ohc1d = da.sum(ohc_basin, axis=(1, 2))
+            ohc1d = da.nansum(ohc_basin, axis=(1, 2))
             ohc1d_total.append(ohc1d)
     return ohc, ohc1d_total