Fixing error in db_init.py