fix db models