Fix import for Django 1.6