diff --git a/dataset.py b/dataset.py
index c4d3e62..38d7cd2 100644
--- a/dataset.py
+++ b/dataset.py
@@ -157,9 +157,6 @@ def create_dataset_from_lists(chunks, vocab, max_len):
     logger.info("  compute domain features")
     domain_features = []
     for ds in tqdm(map(lambda f: f.domain, chunks)):
-        # TODO: fix this correctly
-        # assert min(np.atleast_3d(ds).shape) > 0, f"shape of 0 for {ds}"
-        if not ds: continue
         domain_features.append(np.apply_along_axis(get_domain_features_reduced, 2, np.atleast_3d(ds)))
     domain_features = np.concatenate(domain_features, 0)
     logger.info("  compute flow features")