diff --git a/combo_plugin_imio_townstreet/models.py b/combo_plugin_imio_townstreet/models.py
index 7d7f77d..76a9c90 100644
--- a/combo_plugin_imio_townstreet/models.py
+++ b/combo_plugin_imio_townstreet/models.py
@@ -23,6 +23,7 @@ from combo.data.library import register_cell_class
 
 @register_cell_class
 class TownstreetButton(CellBase):
+    template_name = 'imio_townstreet/button.html'
     default_template_name = 'imio_townstreet/button.html'
 
     class Meta: