diff --git a/runtime/parachains/src/mock.rs b/runtime/parachains/src/mock.rs
index e4da6239e15c..f63588f0e4f3 100644
--- a/runtime/parachains/src/mock.rs
+++ b/runtime/parachains/src/mock.rs
@@ -138,7 +138,7 @@ impl crate::ump::Config for Test {
 	type UmpSink = TestUmpSink;
 	type FirstMessageFactorPercent = FirstMessageFactorPercent;
 	type ExecuteOverweightOrigin = frame_system::EnsureRoot<AccountId>;
-	type WeightInfo = crate::ump::WeightInfo<()>;
+	type WeightInfo = ();
 }
 
 impl crate::hrmp::Config for Test {
diff --git a/runtime/parachains/src/ump/benchmarking.rs b/runtime/parachains/src/ump/benchmarking.rs
index 49bb0a302a99..c20fa782e9bd 100644
--- a/runtime/parachains/src/ump/benchmarking.rs
+++ b/runtime/parachains/src/ump/benchmarking.rs
@@ -83,6 +83,6 @@ frame_benchmarking::benchmarks! {
 
 frame_benchmarking::impl_benchmark_test_suite!(
 	Ump,
-	crate::mock::new_test_ext(Default::default()).build(),
+	crate::mock::new_test_ext(crate::ump::tests::GenesisConfigBuilder::default().build()),
 	crate::mock::Test
 );