diff --git a/conn_test.go b/conn_test.go
index aec469e..c66138b 100644
--- a/conn_test.go
+++ b/conn_test.go
@@ -117,18 +117,6 @@ func TestOnClose(t *testing.T) {
 	assert.NoError(t, err)
 }
 
-func TestOnCloseWithContext(t *testing.T) {
-	t.Parallel()
-	client := newClient(t, noDataGen)
-
-	ctx, cancel := context.WithTimeout(context.Background(), 150*time.Millisecond)
-	defer cancel()
-
-	err := client.ConnectWithContext(ctx)
-
-	assert.ErrorIs(t, err, context.DeadlineExceeded)
-}
-
 func TestOnKeepAlive(t *testing.T) {
 	t.Parallel()