Fixed an import.

This fixes compatibility with Django 1.6