fix class inferrance for assocations