class_name was needed for relation to work