diff --git a/pkg/katautils/config.go b/pkg/katautils/config.go
index ffc5feda8..204a3e5b3 100644
--- a/pkg/katautils/config.go
+++ b/pkg/katautils/config.go
@@ -709,7 +709,10 @@ func updateConfig(configPath string, tomlConf tomlConfig, config *oci.RuntimeCon
 		config.ProxyType = vc.KataBuiltInProxyType
 		config.ShimType = vc.KataBuiltInShimType
 		config.AgentType = vc.KataContainersAgent
-		config.AgentConfig = vc.KataAgentConfig{LongLiveConn: true}
+		config.AgentConfig = vc.KataAgentConfig{
+			LongLiveConn: true,
+			UseVSock:     config.HypervisorConfig.UseVSock,
+		}
 	}
 
 	return nil