diff --git a/lib/xcresult/model_generator.rb b/lib/xcresult/model_generator.rb
index efe5133..29a4f72 100644
--- a/lib/xcresult/model_generator.rb
+++ b/lib/xcresult/model_generator.rb
@@ -40,10 +40,6 @@ def write(destination)
           module XCResult
             module Models
 
-              def self.load_class(class_name)
-                Kernel.const_get("XCResult::Models::\#{class_name}")
-              end
-
         OPEN_MODULE
 
         sorted_types.each do |type|
diff --git a/lib/xcresult/models.gen.rb b/lib/xcresult/models.gen.rb
index 2fa18e1..66d0482 100644
--- a/lib/xcresult/models.gen.rb
+++ b/lib/xcresult/models.gen.rb
@@ -1,5 +1,5 @@
 # This is a generated file. Don't modify this directly!
-# Last generated at: 2021-07-22 07:27:14 UTC
+# Last generated at: 2021-07-25 16:50:48 UTC
 #
 # Name: Xcode Result Types
 # Version: 3.30
@@ -9,10 +9,6 @@
 module XCResult
   module Models
 
-    def self.load_class(class_name)
-      Kernel.const_get("XCResult::Models::#{class_name}")
-    end
-
     # - ActionAbstractTestSummary
     #   * Kind: object
     #   * Properties:
diff --git a/lib/xcresult/models.rb b/lib/xcresult/models.rb
index 106f03c..3f05cc9 100644
--- a/lib/xcresult/models.rb
+++ b/lib/xcresult/models.rb
@@ -5,6 +5,10 @@
 # This file is meant to define extensions to generated models in models.gen.rb.
 module XCResult
   module Models
+    def self.load_class(class_name)
+      Kernel.const_get("XCResult::Models::\#{class_name}")
+    end
+
     class ActionTestableSummary
       def all_tests
         tests.map(&:all_subtests).flatten