diff --git a/deeprvat/deeprvat/associate.py b/deeprvat/deeprvat/associate.py
index e285087f..9dc67537 100644
--- a/deeprvat/deeprvat/associate.py
+++ b/deeprvat/deeprvat/associate.py
@@ -8,7 +8,6 @@
 from pathlib import Path
 from pprint import pprint
 from typing import Dict, List, Optional, Tuple
-
 import click
 import numpy as np
 import pandas as pd
@@ -25,8 +24,6 @@
 from tqdm import tqdm, trange
 import zarr
 import re
-import dask.array as da
-
 import deeprvat.deeprvat.models as deeprvat_models
 from deeprvat.data import DenseGTDataset