Fixed missing association