From a33caa56e7d956a58205c02d60b998957472e0fe Mon Sep 17 00:00:00 2001
From: Alvin Noe Ladines <ladinesalvinnoe@gmail.com>
Date: Tue, 20 Feb 2024 13:57:59 +0100
Subject: [PATCH] Fix tests

---
 tests/test_runschema.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/tests/test_runschema.py b/tests/test_runschema.py
index c6106d7..f23fed3 100644
--- a/tests/test_runschema.py
+++ b/tests/test_runschema.py
@@ -16,12 +16,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import runschema
+from ..runschema.run import Run
+from ..runschema.calculation import Calculation
+from ..runschema.method import Method
+from ..runschema.system import System
 
 
 def test_schema():
-    run = runschema.run.Run()
+    run = Run()
     assert run.m_def.definition_id == 'fc17fdfdcba5aef2c02835a81ed65f95f314b687'
-    run.calculation.append(runschema.calculation.Calculation())
-    run.method.append(runschema.method.Method())
-    run.system.append(runschema.system.System())
+    run.calculation.append(Calculation())
+    run.method.append(Method())
+    run.system.append(System())