diff --git a/src/personalisation/templates/modeladmin/personalisation/segment/index.html b/src/personalisation/templates/modeladmin/personalisation/segment/index.html
index e87bfd7..0c26468 100644
--- a/src/personalisation/templates/modeladmin/personalisation/segment/index.html
+++ b/src/personalisation/templates/modeladmin/personalisation/segment/index.html
@@ -36,21 +36,21 @@
                                                 </ul>
 
                                                 <ul class="inspect segment_rules">
-                                                    {% for rule in segment.time_rules.all %}
+                                                    {% for rule in segment.personalisation_timerule_related.all %}
                                                         <li class="time_rule">   
                                                             These users visit between
                                                             <span>{{ rule.start_time }} and {{ rule.end_time }}</span>
                                                         </li>
                                                     {% endfor %}
 
-                                                    {% for rule in segment.referral_rules.all %}
+                                                    {% for rule in segment.personalisation_referralrule_related.all %}
                                                         <li class="referral_rule">   
                                                             These visits originate from
                                                             <pre>{{ rule.regex_string }}</pre>
                                                         </li>
                                                     {% endfor %}
 
-                                                    {% for rule in segment.visit_count_rules.all %}
+                                                    {% for rule in segment.personalisation_visitcountrule_related.all %}
                                                         <li class="visit_count_rule">   
                                                             These users visited {{ rule.counted_page }}
                                                             <span>{{ rule.get_operator_display }} {{ rule.count }} times</span>
diff --git a/src/personalisation/wagtail_hooks.py b/src/personalisation/wagtail_hooks.py
index ec53309..8692afc 100644
--- a/src/personalisation/wagtail_hooks.py
+++ b/src/personalisation/wagtail_hooks.py
@@ -22,15 +22,6 @@ def register_admin_urls():
     ]
 
 
-class SegmentIndexView(IndexView):
-    def get_context_data(self, **kwargs):
-        context = {
-            'additional_forms': (TimeRuleForm, ReferralRuleForm, VisitCountRuleForm),
-        }
-        context.update(kwargs)
-        return super(SegmentIndexView, self).get_context_data(**context)
-
-
 class SegmentModelAdmin(ModelAdmin):
     """The base model for the Segments administration interface."""
     model = Segment