diff --git a/distribusi/distribusi.py b/distribusi/distribusi.py
index fba0553..91145c7 100644
--- a/distribusi/distribusi.py
+++ b/distribusi/distribusi.py
@@ -2,12 +2,12 @@ import os
 
 import magic
 from PIL import Image
-from distribusi.page_template import html_footer, html_head
-from distribusi.image_templates import (
-    no_description,
-    with_description,
-    with_alttext,
-    full_figure,
+from distribusi.templates.page_template import html_footer, html_head
+from distribusi.templates.image_templates import (
+    image_no_description,
+    image_with_description,
+    image_with_alttext,
+    image_full_figure,
 )
 from distribusi.mappings import CODE_TYPES, FILE_TYPES, SUB_TYPES