diff --git a/dnarbd/segmentmodel.py b/dnarbd/segmentmodel.py
index 784d792c24853dc4441efb7c67ac6ebfac56056d..cf82cd37adfd29d24f862e58abe1fe3627ba7104 100644
--- a/dnarbd/segmentmodel.py
+++ b/dnarbd/segmentmodel.py
@@ -1707,8 +1707,8 @@ class SegmentModel(ArbdModel):
             clusters = hcluster.fclusterdata(data, float(max_basepairs_per_bead)/500, criterion="distance")
             cluster_size = [np.mean(data[clusters == i]) for i in np.unique(clusters)]
         except:
-            clusters = data
-            cluster_size = np.arange(len(data))+1
+            clusters = np.arange(len(data))+1
+            cluster_size = data.flatten()
         for b,c in zip(beads,clusters):
             _assign_bead_type(b, cluster_size[c-1], decimals=order)
 
@@ -1719,8 +1719,8 @@ class SegmentModel(ArbdModel):
             clusters = hcluster.fclusterdata(data, float(max_nucleotides_per_bead)/500, criterion="distance")
             cluster_size = [np.mean(data[clusters == i]) for i in np.unique(clusters)]
         except:
-            clusters = data
-            cluster_size = np.arange(len(data))+1
+            clusters = np.arange(len(data))+1
+            cluster_size = data.flatten()
         for b,c in zip(beads,clusters):
             _assign_bead_type(b, cluster_size[c-1], decimals=order)