Fix little bug in DB class