Exports + Client Register: Make sure to only update state if no err

1 files changed, 6 insertions, 2 deletions

diff --git a/exports/exports.go b/exports/exports.go
index b38d0ff..0779104 100644
--- a/exports/exports.go
+++ b/exports/exports.go
@@ -101,8 +101,6 @@ func Register(
 		return getCError(errors.New("failed to register, a VPN state is already present"))
 	}
 	state := &client.Client{}
-	VPNState = state
-	PStateCallback = stateCallback
 	registerErr := state.Register(
 		C.GoString(name),
 		C.GoString(version),
@@ -110,6 +108,12 @@ func Register(
 		StateCallback,
 		debug != 0,
 	)
+	// Only update the VPN state if we get no error when registering
+	if registerErr == nil {
+		VPNState = state
+		PStateCallback = stateCallback
+		return nil
+	}
 
 	return getCError(registerErr)
 }