Fixed import to work in Django 1.7.