fix models
fix bug relationship in models