diff --git a/gso/cli/netbox.py b/gso/cli/netbox.py
index 00f8f399287412b8cf13964d875e51089bcb9630..d3016443c3870f01cebf521fad0d6a830326c20e 100644
--- a/gso/cli/netbox.py
+++ b/gso/cli/netbox.py
@@ -1,7 +1,7 @@
 import typer
 from pynetbox import RequestError
 
-from gso.services.netbox_client import NetBoxClient, create_device_role, create_device_site
+from gso.services.netbox_client import NetBoxClient
 
 app: typer.Typer = typer.Typer()
 
@@ -18,21 +18,21 @@ def netbox_initial_setup() -> None:
     typer.echo("Connecting to NetBox ...")
 
     try:
-        nbclient = NetBoxClient().connect()
+        nbclient = NetBoxClient()
     except RequestError as e:
         typer.echo(f"Error connecting to NetBox: {e}")
         return
 
     typer.echo("Creating GEANT site ...")
     try:
-        create_device_site(nbclient, "GEANT", "geant")
+        nbclient.create_device_site("GEANT", "geant")
         typer.echo("Site created successfully.")
     except RequestError as e:
         typer.echo(f"Error creating site: {e}")
 
     typer.echo("Creating Router device role ...")
     try:
-        create_device_role(nbclient, "router", "router")
+        nbclient.create_device_role("router", "router")
         typer.echo("Device role created successfully.")
     except RequestError as e:
         typer.echo(f"Error creating device role: {e}")