Support belongs_to with assoc selector