fix associations in models