Fixing error because the database class was not created.