diff --git a/mapping.go b/mapping.go
index b1b65e7..1c239ab 100644
--- a/mapping.go
+++ b/mapping.go
@@ -31,7 +31,7 @@ func (h HostMapping) Add(host, outputUrlStr string) error {
 	case "file":
 		server := http.Server{Handler: http.FileServer(http.Dir(outputUrl.Path))}
 		go server.Serve(pd.listener)
-	case "tcp":
+	case "tcp", "unix":
 		go func(pd ProxyDirection) {
 			for {
 				conn, err := pd.listener.Accept()
@@ -39,18 +39,7 @@ func (h HostMapping) Add(host, outputUrlStr string) error {
 					slog.Debug(err.Error())
 					continue
 				}
-				go handleTcp(conn.(*tls.Conn), pd.Output)
-			}
-		}(pd)
-	case "unix":
-		go func(pd ProxyDirection) {
-			for {
-				conn, err := pd.listener.Accept()
-				if err != nil {
-					slog.Debug(err.Error())
-					continue
-				}
-				go handleUnix(conn.(*tls.Conn), pd.Output)
+				go handleListener(conn.(*tls.Conn), pd.Output)
 			}
 		}(pd)
 	default:
@@ -79,41 +68,12 @@ func handleTlsConn(conn *tls.Conn, hosts HostMapping) error {
 	return nil
 }
 
-func handleTcp(conn *tls.Conn, outputUrl *url.URL) {
-	slog.Debug(fmt.Sprintf("%s -> %s", conn.RemoteAddr(), outputUrl.Host))
-
-	c, err := net.Dial(outputUrl.Scheme, outputUrl.Host)
-	if err != nil {
-		writeError(conn, err)
-		conn.Close()
-		return
-	}
-	defer c.Close()
-
-	wg := new(sync.WaitGroup)
-	wg.Add(2)
-
-	go func() {
-		defer wg.Done()
-
-		_, _ = io.Copy(conn, c)
-	}()
-
-	go func() {
-		defer wg.Done()
-
-		_, _ = io.Copy(c, conn)
-	}()
-
-	wg.Wait()
-}
-
-func handleUnix(conn *tls.Conn, outputUrl *url.URL) {
+func handleListener(conn *tls.Conn, outputUrl *url.URL) {
 	slog.Debug(fmt.Sprintf("%s -> %s", conn.RemoteAddr(), outputUrl.Host+outputUrl.Path))
 
 	c, err := net.Dial(outputUrl.Scheme, outputUrl.Host+outputUrl.Path)
 	if err != nil {
-		writeError(conn, err)
+		fmt.Fprintf(conn, "HTTP/1.1 500 Internal Server Error\r\nConnection: Close\r\nContent-Type: text/plain\r\n\r\n%s", err)
 		conn.Close()
 		return
 	}
@@ -136,7 +96,3 @@ func handleUnix(conn *tls.Conn, outputUrl *url.URL) {
 
 	wg.Wait()
 }
-
-func writeError(w io.Writer, err error) {
-	fmt.Fprintf(w, "HTTP/1.1 500 Internal Server Error\r\nConnection: Close\r\nContent-Type: text/plain\r\n\r\n%s", err)
-}