diff --git a/features/communication.py b/features/communication.py
index 87594fb..2c8b42b 100644
--- a/features/communication.py
+++ b/features/communication.py
@@ -167,8 +167,8 @@ def count_comms(comm_df: pd.DataFrame) -> pd.DataFrame:
         comm_contacts_counts = (
             enumerate_contacts(comm_df)
             .groupby(["participant_id"])
-            .nunique()
-            .rename(columns={"contact_id": "no_contacts"})
+            .nunique()["contact_id"]
+            .rename("no_contacts")
         )
         # Number of communication contacts
         comm_features = comm_counts.join(comm_duration_total)
@@ -199,8 +199,8 @@ def count_comms(comm_df: pd.DataFrame) -> pd.DataFrame:
         comm_contacts_counts = (
             enumerate_contacts(comm_df)
             .groupby(["participant_id"])
-            .nunique()
-            .rename(columns={"contact_id": "no_contacts"})
+            .nunique()["contact_id"]
+            .rename("no_contacts")
         )
         # Number of communication contacts
         comm_features = comm_features.join(comm_contacts_counts)