Fix leftover incorrect association