diff --git a/tests/taint_result_test.py b/tests/taint_result_test.py
index 1b2666e6..65909cbc 100644
--- a/tests/taint_result_test.py
+++ b/tests/taint_result_test.py
@@ -1,5 +1,5 @@
 from mythril.laser.ethereum.taint_analysis import *
-from mythril.laser.ethereum.svm import GlobalState
+from mythril.laser.ethereum.state import GlobalState
 
 
 def test_result_state():
diff --git a/tests/taint_runner_test.py b/tests/taint_runner_test.py
index 03666d2c..d52893c7 100644
--- a/tests/taint_runner_test.py
+++ b/tests/taint_runner_test.py
@@ -2,8 +2,8 @@ import mock
 import pytest
 from pytest_mock import mocker
 from mythril.laser.ethereum.taint_analysis import *
-from mythril.laser.ethereum.svm import GlobalState, Node, Edge, LaserEVM
-from mythril.laser.ethereum.state import MachineState, Account, Environment
+from mythril.laser.ethereum.cfg import Node, Edge
+from mythril.laser.ethereum.state import MachineState, Account, Environment, GlobalState
 
 
 def test_execute_state(mocker):