fix failed tests in model associations