diff --git a/src/features/phone_locations/doryab/main.py b/src/features/phone_locations/doryab/main.py
index 20b06f2d..0dfe03c9 100644
--- a/src/features/phone_locations/doryab/main.py
+++ b/src/features/phone_locations/doryab/main.py
@@ -73,6 +73,13 @@ def cluster_centroids(location_data):
         .reset_index()
     )
 
+    if "index" in centroids.columns:
+        centroids.drop("index", axis=1, inplace=True)
+    if not "local_segment" in centroids.columns:
+        centroids = centroids.assign(local_segment=None)
+    if not "cluster_label" in centroids.columns:
+        centroids = centroids.assign(cluster_label=None)
+
     return centroids
 
 def radius_of_gyration(location_data):