diff --git a/connect/utils_test.go b/connect/utils_test.go
index 39f518b15a92bc9c588558eb4ee3fb4f52997f3f..2d5bb68cfacf229904ce521590c175c035b0f538 100644
--- a/connect/utils_test.go
+++ b/connect/utils_test.go
@@ -83,11 +83,11 @@ func (m mockPartner) Contact() contact.Contact {
 	}
 }
 
-func (m mockPartner) PopSendCypher() (*session.Cypher, error) {
+func (m mockPartner) PopSendCypher() (session.Cypher, error) {
 	return nil, nil
 }
 
-func (m mockPartner) PopRekeyCypher() (*session.Cypher, error) {
+func (m mockPartner) PopRekeyCypher() (session.Cypher, error) {
 	return nil, nil
 }
 
@@ -143,6 +143,22 @@ func newMockConnection(partnerId, myId *id.ID,
 	}
 }
 
+func (m mockConnection) FirstPartitionSize() uint {
+	return 0
+}
+
+func (m mockConnection) SecondPartitionSize() uint {
+	return 0
+}
+
+func (m mockConnection) PartitionSize(payloadIndex uint) uint {
+	return 0
+}
+
+func (m mockConnection) PayloadSize() uint {
+	return 0
+}
+
 func (m mockConnection) Close() error {
 	return nil
 }