Update to work with Django 1.7