diff --git a/probabilistic-analysis/probabilistic_assessment.py b/probabilistic-analysis/probabilistic_assessment.py
index 8867561..38cc0df 100644
--- a/probabilistic-analysis/probabilistic_assessment.py
+++ b/probabilistic-analysis/probabilistic_assessment.py
@@ -486,8 +486,8 @@ def process(beach_name, beach_scenario, n_runs, start_year, end_year,
                 df_out = df_out.assign(segment_gaps=False)
                 for b, p, in zip(segment_gaps['block'],
                                  segment_gaps['profile']):
-                    idx = (df_out['block'] == str(b)) & (df_out['profile']
-                                                         == p)
+                    idx = (str(df_out['block']) == str(b)) & (df_out['profile']
+                                                              == p)
                     df_out.loc[idx, 'segment_gaps'] = True
 
                 # Specify which profiles to plot