diff --git a/lib/annotate/annotate_models.rb b/lib/annotate/annotate_models.rb index dc2901a3..76a6a697 100644 --- a/lib/annotate/annotate_models.rb +++ b/lib/annotate/annotate_models.rb @@ -538,7 +538,7 @@ def annotate(klass, file, header, options = {}) klass.reset_column_information info = get_schema_info(klass, header, options) model_name = klass.name.underscore - table_name = klass.table_name + table_name = klass.table_name if klass.connection_specification_name == ActiveRecord::Base.name model_file_name = File.join(file) annotated = [] @@ -760,7 +760,7 @@ def remove_annotations(options = {}) klass = get_model_class(file) if klass < ActiveRecord::Base && !klass.abstract_class? model_name = klass.name.underscore - table_name = klass.table_name + table_name = klass.table_name if klass.connection_specification_name == ActiveRecord::Base.name model_file_name = file deannotated_klass = true if remove_annotation_of_file(model_file_name, options)