diff --git a/src/wagtail_personalisation/blocks.py b/src/wagtail_personalisation/blocks.py
index 6aaa1fe..fd6d35d 100644
--- a/src/wagtail_personalisation/blocks.py
+++ b/src/wagtail_personalisation/blocks.py
@@ -37,7 +37,7 @@ class PersonalisedStructBlock(blocks.StructBlock):
 
         if value['segment']:
             for segment in user_segments:
-                if segment['id'] == int(value['segment']):
+                if segment.id == int(value['segment']):
                     return super(PersonalisedStructBlock, self).render(
                         value, context)