Fixed issue #1
Import User from models.py