diff --git a/activegraph.gemspec b/activegraph.gemspec
index 32b8e6b79..d41e15fa3 100644
--- a/activegraph.gemspec
+++ b/activegraph.gemspec
@@ -36,6 +36,7 @@ DESCRIPTION
   s.add_dependency('neo4j-ruby-driver', '>= 4.4.1')
   s.add_dependency('orm_adapter', '~> 0.5.0')
   s.add_dependency('sorted_set')
+  s.add_dependency('psych', '>= 4.0.1')
   s.add_development_dependency('guard')
   s.add_development_dependency('guard-rspec')
   s.add_development_dependency('guard-rubocop')
diff --git a/lib/active_graph/tasks/migration.rake b/lib/active_graph/tasks/migration.rake
index 0cbb6a787..b3d65eeac 100644
--- a/lib/active_graph/tasks/migration.rake
+++ b/lib/active_graph/tasks/migration.rake
@@ -83,7 +83,7 @@ COMMENT
 
       args.with_defaults(remove_missing: false)
 
-      schema_data = YAML.safe_load(File.read(SCHEMA_YAML_PATH), [Symbol])
+      schema_data = YAML.safe_load(File.read(SCHEMA_YAML_PATH), permitted_classes: [Symbol])
 
       ActiveGraph::Base.subscribe_to_query(&method(:puts))