diff --git a/src/pymorize/cmorizer.py b/src/pymorize/cmorizer.py
index be6c8432..a75e27d1 100644
--- a/src/pymorize/cmorizer.py
+++ b/src/pymorize/cmorizer.py
@@ -2,6 +2,7 @@
 
 import dask
 import pandas as pd
+import xarray as xr
 import questionary
 import yaml
 from dask.distributed import Client