diff --git a/portalnetwork/state/network_test.go b/portalnetwork/state/network_test.go
index ee1371d0d3..965db5c3fc 100644
--- a/portalnetwork/state/network_test.go
+++ b/portalnetwork/state/network_test.go
@@ -82,7 +82,8 @@ func TestValidateContractStorage(t *testing.T) {
 			header: tt.BlockHeader,
 		}
 		server.RegisterName("portal", api)
-		bn := NewStateNetwork(nil, server)
+		client := rpc.DialInProc(server)
+		bn := NewStateNetwork(nil, client)
 		err = bn.validateContent(hexutil.MustDecode(tt.ContentKey), hexutil.MustDecode(tt.ContentValueOffer))
 		require.NoError(t, err)
 	}
@@ -98,7 +99,8 @@ func TestValidateContractByte(t *testing.T) {
 			header: tt.BlockHeader,
 		}
 		server.RegisterName("portal", api)
-		bn := NewStateNetwork(nil, server)
+		client := rpc.DialInProc(server)
+		bn := NewStateNetwork(nil, client)
 		err = bn.validateContent(hexutil.MustDecode(tt.ContentKey), hexutil.MustDecode(tt.ContentValueOffer))
 		require.NoError(t, err)
 	}