diff --git a/R/sessionLogger.R b/R/sessionLogger.R
index f671f86..89535c0 100644
--- a/R/sessionLogger.R
+++ b/R/sessionLogger.R
@@ -99,7 +99,7 @@ SessionLoggerDB <- R6::R6Class(
   inherit = SessionLogger,
   public = list(
     initialize = function(db_client) {
-      if (is.null(conn) || !R6::is.R6(conn) || !inherits(conn, "DBClient")) {
+      if (is.null(db_client) || !inherits(db_client, "DBClient")) {
         stop("DBClient instance required!")
       }
       private$db_client <- db_client
@@ -111,15 +111,15 @@ SessionLoggerDB <- R6::R6Class(
     logSessionEvent = function(
       event, session = shiny::getDefaultReactiveDomain()
     ) {
-      db_client$connect()
+      private$db_client$getConnection()$connect()
       on.exit({
-        db_client$disconnect()
+        private$db_client$getConnection()$disconnect()
       })
 
       event_row <- tibble::tibble(
         app_name = basename(getwd()),
         app_pid = Sys.getpid(),
-        client_ip = getClientIp(session),
+        client_ip = private$getClientIp(session),
         user_id = session$userData$user_id,
         session_token = session$token,
         !!event := lubridate::now()