Quick DB class fix