Fixed import on models.py

Forgot to import six.