add compatibility with STI in decorate_association

The simplest solution with in decorate_association STI