better support for class_name on associations