diff --git a/machine_learning/pipeline.py b/machine_learning/pipeline.py
index f5c35ae..f08d2cf 100644
--- a/machine_learning/pipeline.py
+++ b/machine_learning/pipeline.py
@@ -50,6 +50,8 @@ class SensorFeatures:
     def get_sensor_data(self, data_type) -> pd.DataFrame:
         if data_type == "proximity":
             return self.df_proximity
+        elif data_type == "communication":
+            return self.df_calls_sms
         else:
             raise KeyError("This data type has not been implemented.")