fix db class