diff --git a/client_test.go b/client_test.go
index a2aea93..3e1f180 100644
--- a/client_test.go
+++ b/client_test.go
@@ -22,7 +22,15 @@ func TestAPI(t *testing.T) {
 	httpClient := http.DefaultClient
 	defer httpClient.CloseIdleConnections()
 
-	vokiClient := voki.New(httpClient, "localhost", cfg.API.Key, cfg.API.Port, "http")
+	var protocol string
+	switch cfg.API.TLS.Enabled {
+	case true:
+		protocol = "https"
+	case false:
+		protocol = "http"
+	}
+
+	vokiClient := voki.New(httpClient, "localhost", cfg.API.Key, cfg.API.Port, protocol)
 	apiClient := APIClient{vokiClient}
 
 	t.Run("get API health", func(t *testing.T) {