diff --git a/share/templates.d/99-generic/arm.tmpl b/share/templates.d/99-generic/arm.tmpl
index d7372daf..99fa39a0 100644
--- a/share/templates.d/99-generic/arm.tmpl
+++ b/share/templates.d/99-generic/arm.tmpl
@@ -11,6 +11,8 @@ LORAXDIR="usr/share/lorax/"
 # keep a comma-delimited list of platforms installed to add to .treeinfo
 platforms = ""
 delimiter = ''
+
+from os.path import basename
 %>
 
 mkdir images