From 3386a0171a451d2cbf78796aea55655afe422e0b Mon Sep 17 00:00:00 2001
From: Malte Kliemann <mail@maltekliemann.com>
Date: Sun, 7 Jul 2024 23:47:31 +0200
Subject: [PATCH] Fix tests and imports

---
 zrml/neo-swaps/src/mock.rs          | 2 +-
 zrml/neo-swaps/src/tests/buy.rs     | 1 +
 zrml/prediction-markets/src/mock.rs | 3 +--
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/zrml/neo-swaps/src/mock.rs b/zrml/neo-swaps/src/mock.rs
index 62eb2b7c7..f066a31cd 100644
--- a/zrml/neo-swaps/src/mock.rs
+++ b/zrml/neo-swaps/src/mock.rs
@@ -306,7 +306,7 @@ impl frame_system::Config for Runtime {
     type Hashing = BlakeTwo256;
     type Lookup = IdentityLookup<Self::AccountId>;
     type Nonce = u64;
-    type MaxConsumers = frame_support::traits::ConstU32<16>;
+    type MaxConsumers = ConstU32<16>;
     type OnKilledAccount = ();
     type OnNewAccount = ();
     type RuntimeOrigin = RuntimeOrigin;
diff --git a/zrml/neo-swaps/src/tests/buy.rs b/zrml/neo-swaps/src/tests/buy.rs
index 9ecd703bb..cb7669348 100644
--- a/zrml/neo-swaps/src/tests/buy.rs
+++ b/zrml/neo-swaps/src/tests/buy.rs
@@ -16,6 +16,7 @@
 // along with Zeitgeist. If not, see <https://www.gnu.org/licenses/>.
 
 use super::*;
+use sp_runtime::{DispatchError, TokenError};
 use test_case::test_case;
 
 // Example taken from
diff --git a/zrml/prediction-markets/src/mock.rs b/zrml/prediction-markets/src/mock.rs
index a98030ed7..e05a9b232 100644
--- a/zrml/prediction-markets/src/mock.rs
+++ b/zrml/prediction-markets/src/mock.rs
@@ -28,7 +28,6 @@ use frame_support::{
     traits::{Everything, NeverEnsureOrigin, OnFinalize, OnInitialize},
 };
 use frame_system::{mocking::MockBlock, EnsureRoot, EnsureSignedBy};
-use parity_scale_codec::Encode;
 use sp_arithmetic::per_things::Percent;
 use sp_runtime::{
     traits::{BlakeTwo256, ConstU32, IdentityLookup},
@@ -247,7 +246,7 @@ impl frame_system::Config for Runtime {
     type Hashing = BlakeTwo256;
     type Lookup = IdentityLookup<Self::AccountId>;
     type Nonce = u64;
-    type MaxConsumers = frame_support::traits::ConstU32<16>;
+    type MaxConsumers = ConstU32<16>;
     type OnKilledAccount = ();
     type OnNewAccount = ();
     type RuntimeOrigin = RuntimeOrigin;