diff --git a/Cargo.lock b/Cargo.lock
index d2011def8..54e39bd68 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -16948,7 +16948,7 @@ dependencies = [
  "parity-scale-codec",
  "polkadot-runtime-parachains",
  "rococo-runtime",
- "rococo-runtime-constants",
+ "westend-runtime",
 ]
 
 [[package]]
diff --git a/runtime/relay-encoder/Cargo.toml b/runtime/relay-encoder/Cargo.toml
index e59f0134a..0c6230fb0 100644
--- a/runtime/relay-encoder/Cargo.toml
+++ b/runtime/relay-encoder/Cargo.toml
@@ -19,7 +19,7 @@ parity-scale-codec = { workspace = true, features = [ "derive" ] }
 [dev-dependencies]
 polkadot-runtime-parachains = { workspace = true }
 rococo-runtime = { workspace = true }
-rococo-runtime-constants = { workspace = true }
+westend-runtime = { workspace = true }
 
 [features]
 default = [
@@ -29,6 +29,6 @@ std = [
 	"cumulus-primitives-core/std",
 	"parity-scale-codec/std",
 	"polkadot-runtime-parachains/std",
-	"rococo-runtime-constants/std",
 	"rococo-runtime/std",
+	"westend-runtime/std",
 ]
diff --git a/runtime/relay-encoder/src/westend.rs b/runtime/relay-encoder/src/westend.rs
index 61db20ee1..f5a014f71 100644
--- a/runtime/relay-encoder/src/westend.rs
+++ b/runtime/relay-encoder/src/westend.rs
@@ -33,12 +33,11 @@ pub enum OnDemandAssignmentProviderCall {
     },
 }
 
-// TODO: uncomment tests after polkadot 1.8.0 upgrade
-/*
 #[cfg(test)]
 mod tests {
-    use super::*;
-    use polkadot_runtime_parachains::assigner_on_demand as parachains_assigner_on_demand;
+    use {
+        super::*, polkadot_runtime_parachains::assigner_on_demand as parachains_assigner_on_demand,
+    };
 
     #[test]
     fn encode_place_order_allow_death() {
@@ -61,4 +60,3 @@ mod tests {
         assert_eq!(call.encode(), call2.encode());
     }
 }
-*/