Corrected models associations