1 files changed, 2 insertions, 3 deletions
diff --git a/test/session_debugger/session_debugger.c b/test/session_debugger/session_debugger.c
index b54d739..80f5302 100644
--- a/test/session_debugger/session_debugger.c
+++ b/test/session_debugger/session_debugger.c
@@ -287,9 +287,8 @@ struct stellar_module *session_debugger_on_init(struct stellar_module_manager *m
 {
     assert(mod_mgr);
 
-    struct stellar_module *sess_mgr_mod = stellar_module_manager_get_module(mod_mgr, SESSION_MANAGER_MODULE_NAME);
-    assert(sess_mgr_mod);
-    struct session_manager *sess_mgr = stellar_module_get_ctx(sess_mgr_mod);
+
+    struct session_manager *sess_mgr = stellar_module_get_session_manager(mod_mgr);
     assert(sess_mgr);
     struct logger *logger = stellar_module_manager_get_logger(mod_mgr);
     assert(logger);