Fix to import name in database.py