From 9c1251c3ce71e7a14b5933c1f1e4b3fcebdb85c5 Mon Sep 17 00:00:00 2001
From: claravanstaden <Cats 4 life!>
Date: Tue, 24 Oct 2023 22:36:55 +0200
Subject: [PATCH] fixing tests

---
 smoketest/tests/create_channel.rs | 12 ++++++------
 smoketest/tests/register_token.rs |  4 ++--
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/smoketest/tests/create_channel.rs b/smoketest/tests/create_channel.rs
index 4cd43b3c50..7ec08a6d69 100644
--- a/smoketest/tests/create_channel.rs
+++ b/smoketest/tests/create_channel.rs
@@ -1,17 +1,17 @@
 use snowbridge_smoketest::contracts::i_gateway::ChannelCreatedFilter;
 use snowbridge_smoketest::helper::*;
 use snowbridge_smoketest::parachains::bridgehub::api::ethereum_control::events::CreateChannel;
-use snowbridge_smoketest::xcm::construct_xcm_message;
+use snowbridge_smoketest::xcm::construct_xcm_message_with_fee;
 
 #[tokio::test]
 async fn create_channel() {
     let test_clients = initial_clients().await.expect("initialize clients");
 
-    let message = construct_xcm_message(
-        construct_create_channel_call(&test_clients.bridge_hub_client)
-            .await
-            .expect("construct innner call."),
-    );
+    let encoded_call = construct_create_channel_call(&test_clients.bridge_hub_client)
+        .await
+        .expect("construct innner call.");
+
+    let message = construct_xcm_message_with_fee(encoded_call).await;
 
     let result = send_xcm_transact(&test_clients.template_client, message)
         .await
diff --git a/smoketest/tests/register_token.rs b/smoketest/tests/register_token.rs
index cb92cef62d..a6ee934751 100644
--- a/smoketest/tests/register_token.rs
+++ b/smoketest/tests/register_token.rs
@@ -13,13 +13,13 @@ use snowbridge_smoketest::{
     contracts::{i_gateway, weth9},
     parachains::assethub::api::{
         foreign_assets::events::Created,
-        runtime_types::staging_xcm::v3::{
+        runtime_types::staging_xcm::v3::multilocation::MultiLocation,
+        runtime_types::xcm::v3::{
             junction::{
                 Junction::{AccountKey20, GlobalConsensus},
                 NetworkId,
             },
             junctions::Junctions::X3,
-            multilocation::MultiLocation,
         },
     },
 };