diff --git a/go/pserver/client_test.go b/go/pserver/client_test.go
index 6ecf1fa08a02ed2ce04fae0903cebd46a7b768a4..4a62ae88a4a8863c29c3fe23db0648f3e1141eaf 100644
--- a/go/pserver/client_test.go
+++ b/go/pserver/client_test.go
@@ -31,7 +31,7 @@ func init() {
 		port[i] = p
 
 		go func(l net.Listener) {
-			s, err := pserver.NewService("", time.Second*5)
+			s, err := pserver.NewService("", 1, time.Second*5)
 			if err != nil {
 				panic(err)
 			}
diff --git a/go/pserver/service_test.go b/go/pserver/service_test.go
index f317535592165b921491120888badd30c6795c12..1d84f15d78afff90e9b5fa40e97a88cfdc54ee91 100644
--- a/go/pserver/service_test.go
+++ b/go/pserver/service_test.go
@@ -10,7 +10,7 @@ import (
 )
 
 func TestFull(t *testing.T) {
-	s, err := pserver.NewService("", time.Second*5)
+	s, err := pserver.NewService("", 1, time.Second*5)
 	if err != nil {
 		t.Error(err)
 	}
@@ -75,7 +75,7 @@ func TestFull(t *testing.T) {
 }
 
 func TestMultipleInit(t *testing.T) {
-	s, err := pserver.NewService("", time.Second*5)
+	s, err := pserver.NewService("", 1, time.Second*5)
 	if err != nil {
 		t.Error(err)
 	}
@@ -91,7 +91,7 @@ func TestMultipleInit(t *testing.T) {
 }
 
 func TestUninitialized(t *testing.T) {
-	s, err := pserver.NewService("", time.Second*5)
+	s, err := pserver.NewService("", 1, time.Second*5)
 	err = s.SendGrad(pserver.Gradient{}, nil)
 	if err.Error() != pserver.Uninitialized {
 		t.FailNow()
@@ -99,7 +99,7 @@ func TestUninitialized(t *testing.T) {
 }
 
 func TestBlockUntilInitialized(t *testing.T) {
-	s, err := pserver.NewService("", time.Second*5)
+	s, err := pserver.NewService("", 1, time.Second*5)
 	if err != nil {
 		t.Error(err)
 	}