1 files changed, 9 insertions, 2 deletions
diff --git a/pkg/bot/bot.go b/pkg/bot/bot.go
index 264e328..a0c0808 100644
--- a/pkg/bot/bot.go
+++ b/pkg/bot/bot.go
@@ -1,6 +1,7 @@
 package bot
 
 import (
+	"context"
 	"fmt"
 	"jinx/pkg/discord"
 )
@@ -12,11 +13,17 @@ func Start() error {
 
 	client := discord.NewClient(TOKEN)
 
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
 	fmt.Println("connecting..")
+	if err := client.Connect(ctx); err != nil {
+		return err
+	}
 
-	err := client.Connect()
+	fmt.Println("connected..")
 
-	if err != nil {
+	if err := client.Disconnect(); err != nil {
 		return err
 	}