diff --git a/driver/state/state_test.go b/driver/state/state_test.go
index 3ca7d99b3..aa4066065 100644
--- a/driver/state/state_test.go
+++ b/driver/state/state_test.go
@@ -14,16 +14,25 @@ import (
 
 type DriverStateTestSuite struct {
 	testutils.ClientTestSuite
-	s *State
+	ctx    context.Context
+	cancel context.CancelFunc
+	s      *State
 }
 
 func (s *DriverStateTestSuite) SetupTest() {
 	s.ClientTestSuite.SetupTest()
-	state, err := New(context.Background(), s.RPCClient)
+	s.ctx, s.cancel = context.WithCancel(context.Background())
+	state, err := New(s.ctx, s.RPCClient)
 	s.Nil(err)
 	s.s = state
 }
 
+func (s *DriverStateTestSuite) TearDownTest() {
+	if s.ctx.Err() == nil {
+		s.cancel()
+	}
+}
+
 func (s *DriverStateTestSuite) TestGetL1Head() {
 	l1Head := s.s.GetL1Head()
 	s.NotNil(l1Head)
@@ -34,6 +43,7 @@ func (s *DriverStateTestSuite) TestGetHeadBlockID() {
 }
 
 func (s *DriverStateTestSuite) TestClose() {
+	s.cancel()
 	s.NotPanics(s.s.Close)
 }