fix for polymorphic work association