diff --git a/mlclouds/p_fun.py b/mlclouds/p_fun.py
index 804d63c..8cf5144 100644
--- a/mlclouds/p_fun.py
+++ b/mlclouds/p_fun.py
@@ -36,6 +36,7 @@ def decode_sky_type(cloud_type):
     """Decode integer cloud types as strings."""
     if 'cloud_type' in cloud_type:
         df = cloud_type.copy()
+        df = df.rename({'cloud_type': 'flag'}, axis=1)
     else:
         df = pd.DataFrame(columns=['flag'], data=cloud_type)
     df.loc[df['flag'] == 0, 'flag'] = 'clear'