CLI: Deregister to save configuration

1 files changed, 3 insertions, 2 deletions

diff --git a/cli/main.go b/cli/main.go
index a9a1167..d5c3aa8 100644
--- a/cli/main.go
+++ b/cli/main.go
@@ -1,9 +1,9 @@
 package main
 
 import (
+	"errors"
 	"flag"
 	"fmt"
-	"errors"
 	"os"
 	"os/exec"
 	"strings"
@@ -33,7 +33,6 @@ func writeGraph(filename string) error {
 	graph := state.GenerateGraph()
 
 	f, err := os.Create(filename)
-
 	if err != nil {
 		return errors.New(fmt.Sprintf("Failed to create file %s with error %v", filename, err))
 	}
@@ -76,6 +75,8 @@ func main() {
 
 		fmt.Println(config)
 
+		state.Deregister()
+
 		return
 	}