fix associations bug