Fix polymorphic assosication problem