Updating to work with Django 1.6