diff --git a/insomniac/safely_runner.py b/insomniac/safely_runner.py index 0efbac9..7f8ac9d 100644 --- a/insomniac/safely_runner.py +++ b/insomniac/safely_runner.py @@ -2,8 +2,6 @@ from socket import timeout from insomniac.device_facade import DeviceFacade -from insomniac.extra_features.actions_impl import airplane_mode_on_off -from insomniac.extra_features.utils import new_identity from insomniac.navigation import navigate, LanguageChangedException from insomniac.sleeper import sleeper from insomniac.utils import * @@ -42,6 +40,9 @@ def wrapper(*args, **kwargs): def run_registration_safely(device_wrapper): + from insomniac.extra_features.actions_impl import airplane_mode_on_off + from insomniac.extra_features.utils import new_identity + def actual_decorator(func): def wrapper(*args, **kwargs): try: