Add missing associations