Fixed missing database connection error