Fix DB model