lib/torque/postgresql/auxiliary_statement.rb in torque-postgresql-0.1.4 vs lib/torque/postgresql/auxiliary_statement.rb in torque-postgresql-0.1.5

@@ -167,12 +167,12 @@
             foreign_key = (polymorphic.present? ? polymorphic : base_name)
             foreign_key = foreign_key.to_s.foreign_key
             if query.columns_hash.key?(foreign_key)
               joins_projections(base.primary_key => foreign_key)
               if polymorphic.present?
-                foreign_type = project(foreign_key.gsub(/_id$/, '_type'), query_table)
-                @selected_attributes << foreign_type
-                @join_attributes << foreign_type.eq(base_name)
+                foreign_type = foreign_key.gsub(/_id$/, '_type')
+                @selected_attributes << project(foreign_type, query_table)
+                @join_attributes << project(foreign_type).eq(base_name)
               end
             end
           end
       end