diff --git a/src/personalisation/wagtail_hooks.py b/src/personalisation/wagtail_hooks.py
index 6d7afa4..2175482 100644
--- a/src/personalisation/wagtail_hooks.py
+++ b/src/personalisation/wagtail_hooks.py
@@ -9,7 +9,7 @@ from wagtail.contrib.modeladmin.helpers import ButtonHelper
 from wagtail.contrib.modeladmin.options import ModelAdmin, modeladmin_register
 from wagtail.wagtailcore import hooks
 
-from .models import Segment
+from personalisation.models import Segment
 
 
 @hooks.register('register_admin_urls')
@@ -33,4 +33,4 @@ class SegmentModelAdmin(ModelAdmin):
     add_to_settings_menu = False
     button_helper_class = SegmentButtonHelper
 
-modeladmin_register(SegmentModelAdmin)
\ No newline at end of file
+modeladmin_register(SegmentModelAdmin)