Fixed a missing import error in auth.py