diff --git a/go/pserver/internal/connection/conn.go b/go/connection/conn.go
similarity index 100%
rename from go/pserver/internal/connection/conn.go
rename to go/connection/conn.go
diff --git a/go/pserver/client.go b/go/pserver/client.go
index f8bd0aa59f..4f35141a9f 100644
--- a/go/pserver/client.go
+++ b/go/pserver/client.go
@@ -6,7 +6,7 @@ import (
 	"sort"
 	"time"
 
-	"github.com/PaddlePaddle/Paddle/go/pserver/internal/connection"
+	"github.com/PaddlePaddle/Paddle/go/connection"
 )
 
 // TODO(helin): add RPC call retry logic