diff --git a/main.go b/main.go
index 799f544..7e8a396 100644
--- a/main.go
+++ b/main.go
@@ -38,7 +38,7 @@ func main() {
 	}
 
 	traceSession, err := cluster.CreateSession()
-	defer session.Close()
+	defer traceSession.Close()
 
 	if err != nil {
 		log.Critical(err)