fix belongs_to with lambda