lib/tapioca/dsl/compilers/state_machines.rb in tapioca-0.9.4 vs lib/tapioca/dsl/compilers/state_machines.rb in tapioca-0.10.0

@@ -158,12 +158,16 @@
             klass.create_include(instance_module_name)
             klass.create_extend(class_module_name)
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          all_classes.select { |mod| mod < ::StateMachines::InstanceMethods }
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            all_classes.select { |mod| mod < ::StateMachines::InstanceMethods }
+          end
         end
 
         private
 
         sig { params(machine: ::StateMachines::Machine).returns(String) }