Fix missing association