Merge branch 'mb/features' into develop

1 files changed, 5 insertions, 5 deletions

diff --git a/test/apps/test_migrate.py b/test/apps/test_migrate.py
index 957509a..230c032 100644
--- a/test/apps/test_migrate.py
+++ b/test/apps/test_migrate.py
@@ -13,7 +13,7 @@ import unittest
 import unittest.mock
 
 # bsie imports
-from bsfs.schema import Schema
+from bsfs import schema
 
 # objects to test
 from bsfs.apps.migrate import main
@@ -33,21 +33,21 @@ class TestMigrate(unittest.TestCase):
 
         # read schema from file
         with open(schema_1) as ifile:
-            target = Schema.from_string(ifile.read())
+            target = schema.from_string(ifile.read())
         graph = main([config, schema_1])
         self.assertTrue(target <= graph.schema)
 
         # read schema from multiple files
         with open(schema_1) as ifile:
-            target = Schema.from_string(ifile.read())
+            target = schema.from_string(ifile.read())
         with open(schema_2) as ifile:
-            target = target + Schema.from_string(ifile.read())
+            target = target + schema.from_string(ifile.read())
         graph = main([config, schema_1, schema_2])
         self.assertTrue(target <= graph.schema)
 
         # read schema from stdin
         with open(schema_1, 'rt') as ifile:
-            target = Schema.from_string(ifile.read())
+            target = schema.from_string(ifile.read())
         with open(schema_1, 'rt') as ifile:
             with unittest.mock.patch('sys.stdin', ifile):
                 graph = main([config])