diff --git a/session_test.go b/session_test.go
index b02dbdd..9aca780 100644
--- a/session_test.go
+++ b/session_test.go
@@ -91,6 +91,7 @@ func TestGetDieCh(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer ss.Close()
 	dieCh := ss.GetDieCh()
 	go func() {
 		select {