ms_driver: Catch up with the new MS baseclass
Change-Id: Id8de8b3ee76b515c49e5ea52acaa326a2283b0e1
diff --git a/src/osmo_ms_driver/__main__.py b/src/osmo_ms_driver/__main__.py
index 24acd61..d753897 100644
--- a/src/osmo_ms_driver/__main__.py
+++ b/src/osmo_ms_driver/__main__.py
@@ -23,7 +23,7 @@
from .starter import BinaryOptions
from .test_support import imsi_ki_gen
from osmo_gsm_tester import log, util
-from osmo_gsm_tester import ms
+from osmo_gsm_tester import ms_osmo_mobile
# System modules
from datetime import timedelta
@@ -97,7 +97,7 @@
'ki': ki,
'auth_algo': 'comp128v1',
}
- test.subscriber_add(ms.MS("ms_%d" % i, conf))
+ test.subscriber_add(ms_osmo_mobile.MSOsmoMobile("ms_%d" % i, conf))
atexit.register(test.stop_all)