Update deprecated import

Required for Django 1.6