fix  model associations