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 f8bd0aa59f30ec7e2b2d318929af96135d3128ed..4f35141a9f725f11fe232d46e7904a5322d977b0 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