diff --git a/climax/core/dataset.py b/climax/core/dataset.py
index 5e79106243fe6472716aecb0065cd67c96b7b5ce..b521bbebcbd6140e6d76ae620cd4988a4ccf0edd 100644
--- a/climax/core/dataset.py
+++ b/climax/core/dataset.py
@@ -142,7 +142,9 @@ class EoDataset(torch.utils.data.Dataset):
             alg='Horn').ReadAsArray().astype(np.float32)
 
         # read digital elevation model
-        dem = img2np(dem).astype(np.float32)
+        dem = da.asarray(img2np(dem).astype(np.float32))
+        slope = da.asarray(slope)
+        aspect = da.asarray(aspect)
 
         # digital elevation model features: elevation, slope and aspect
         dem_vars = {