diff --git a/cmd/csi-attacher/main.go b/cmd/csi-attacher/main.go
index 72953b0b7d..4a28f20255 100644
--- a/cmd/csi-attacher/main.go
+++ b/cmd/csi-attacher/main.go
@@ -105,7 +105,7 @@ func main() {
 	factory := informers.NewSharedInformerFactory(clientset, *resync)
 	var handler controller.Handler
 	// Connect to CSI.
-	csiConn, err := connection.Connect(*csiAddress)
+	csiConn, err := connection.Connect(*csiAddress, connection.OnConnectionLoss(connection.ExitOnConnectionLoss()))
 	if err != nil {
 		klog.Error(err.Error())
 		os.Exit(1)