diff --git a/cli_test.go b/cli_test.go
index 49a6255..f29d466 100644
--- a/cli_test.go
+++ b/cli_test.go
@@ -5,7 +5,6 @@ import (
 	"log"
 	"strings"
 	"testing"
-	"gopkg.in/urfave/cli.v2"
 	"errors"
 )
 
@@ -84,6 +83,14 @@ func TestStart(t *testing.T) {
 	}
 }
 
+func TestSetup(t *testing.T) {
+	m := mockRealize{}
+	mockResponse = nil
+	if err := m.setup(); err != nil{
+		t.Fatal("Unexpected error")
+	}
+}
+
 func TestClean(t *testing.T) {
 	m := mockRealize{}
 	mockResponse = nil