Updated assoc_class key name