diff --git a/client/mockServer_test.go b/client/mockServer_test.go
index 73079cacac1c399c1d5ec9af4d0106cb311126cd..3ec1f5dc6fa4d717539d0b0726a4f52909a404e5 100644
--- a/client/mockServer_test.go
+++ b/client/mockServer_test.go
@@ -15,7 +15,6 @@ import (
 	pb "gitlab.com/elixxir/comms/mixmessages"
 	"gitlab.com/elixxir/comms/testutils"
 	"gitlab.com/xx_network/comms/connect"
-	"gitlab.com/xx_network/primitives/id"
 	"gitlab.com/xx_network/primitives/ndf"
 	"sync"
 )
@@ -53,8 +52,8 @@ func getNextAddress() string {
 type MockRegistration struct {
 }
 
-func (s *MockRegistration) RegisterNode(NodeID *id.ID,
-	NodeTLSCert, GatewayTLSCert, RegistrationCode, Addr, Addr2 string) error {
+func (s *MockRegistration) RegisterNode(salt []byte, serverAddr, serverTlsCert, gatewayAddr,
+	gatewayTlsCert, registrationCode string) error {
 	return nil
 }
 
@@ -85,8 +84,8 @@ func (s *MockRegistration) CheckRegistration(msg *pb.RegisteredNodeCheck) (*pb.R
 type MockRegistrationError struct {
 }
 
-func (s *MockRegistrationError) RegisterNode(NodeID *id.ID,
-	NodeTLSCert, GatewayTLSCert, RegistrationCode, Addr, Addr2 string) error {
+func (s *MockRegistrationError) RegisterNode(salt []byte, serverAddr, serverTlsCert, gatewayAddr,
+	gatewayTlsCert, registrationCode string) error {
 	return nil
 }
 
diff --git a/node/register_test.go b/node/register_test.go
index 2bee3309ee1f1a6174336132496e6a19d0b579c0..f9f6f0642127571391629c61249d9e5de76a48a4 100644
--- a/node/register_test.go
+++ b/node/register_test.go
@@ -32,7 +32,7 @@ func TestSendNodeRegistration(t *testing.T) {
 		t.Errorf("Unable to call NewHost: %+v", err)
 	}
 
-	msgs := &pb.NodeRegistration{ID: testId.Bytes()}
+	msgs := &pb.NodeRegistration{Salt: testId.Bytes()}
 	err = server.SendNodeRegistration(host, msgs)
 	if err != nil {
 		t.Errorf("SendNodeTopology: Error received: %s", err)