diff --git a/lib/state_machines/yard/handlers/machine.rb b/lib/state_machines/yard/handlers/machine.rb
index c34e0ca..b532074 100644
--- a/lib/state_machines/yard/handlers/machine.rb
+++ b/lib/state_machines/yard/handlers/machine.rb
@@ -117,7 +117,7 @@ def instance_attributes
         # Gets the type of ORM integration being used based on the list of
         # ancestors (including mixins)
         def integration
-          @integration ||= Integrations.match_ancestors(namespace.inheritance_tree(true).map { |ancestor| ancestor.path })
+          @integration ||= StateMachines::Integrations.match_ancestors(namespace.inheritance_tree(true).map { |ancestor| ancestor.path })
         end
 
         # Gets the class type being used to define states.  Default is "Symbol".