lib/graphiti_gql/schema/fields/stats.rb in graphiti_gql-0.2.1 vs lib/graphiti_gql/schema/fields/stats.rb in graphiti_gql-0.2.2

@@ -27,11 +27,11 @@
 
         def build_stat_class
           name = Registry.instance.key_for(@resource)
           stat_graphql_name = "#{name}Stats"
           return Registry.instance[stat_graphql_name][:type] if Registry.instance[stat_graphql_name]
-          klass = Class.new(GraphQL::Schema::Object)
+          klass = Class.new(Schema.base_object)
           klass.graphql_name(stat_graphql_name)
           @resource.stats.each_pair do |name, config|
             calc_class = build_calc_class(stat_graphql_name, name, config.calculations.keys)
             klass.field name, calc_class, null: false
           end
@@ -39,10 +39,10 @@
           klass
         end
     
         def build_calc_class(stat_graphql_name, stat_name, calculations)
           name = "#{stat_graphql_name}#{stat_name}Calculations"
-          klass = Class.new(GraphQL::Schema::Object)
+          klass = Class.new(Schema.base_object)
           klass.graphql_name(name)
           calculations.each do |calc|
             klass.field calc, Float, null: false
           end
           klass
\ No newline at end of file