Fix to template, weird import problem in models.