diff --git a/aliby/utils/imageViewer.py b/aliby/utils/imageViewer.py
index 59ea073471522ea8f9108863ec6bc880a621172f..de766d8bfe6d204fec0df926f217faa9cec43b92 100644
--- a/aliby/utils/imageViewer.py
+++ b/aliby/utils/imageViewer.py
@@ -23,7 +23,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import seaborn as sns
 import yaml
-from agora.io.cells import CellsLinear as Cells
+from agora.io.cells import Cells
 from agora.io.writer import load_attributes
 from PIL import Image
 from skimage.morphology import dilation
diff --git a/extraction/core/extractor.py b/extraction/core/extractor.py
index 285cb7721c84995f8e2089575f92436e65cdac07..6d5a4510d586c0d3e89f73ee687ae456bb9f82d6 100644
--- a/extraction/core/extractor.py
+++ b/extraction/core/extractor.py
@@ -6,7 +6,7 @@ import h5py
 import numpy as np
 import pandas as pd
 from agora.abc import ParametersABC, ProcessABC
-from agora.io.cells import CellsLinear
+from agora.io.cells import Cells
 from agora.io.writer import Writer, load_attributes
 
 from aliby.tile.tiler import Tiler
@@ -338,7 +338,7 @@ class Extractor(ProcessABC):
         ch_tree = {ch: v for ch, v in tree.items() if ch != "general"}
         tree_chs = (*ch_tree,)
 
-        cells = CellsLinear(self.local)
+        cells = Cells(self.local)
 
         # labels
         if labels is None: