diff --git a/contract-rs/Cargo.toml b/contract-rs/Cargo.toml
index 443a7ca..6006d3d 100644
--- a/contract-rs/Cargo.toml
+++ b/contract-rs/Cargo.toml
@@ -13,11 +13,11 @@ crate-type = ["cdylib", "rlib"]
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [dependencies]
-near-sdk = { version = "5.1.0", features = ["legacy"] }
+near-sdk = "5.4.0"
 
 [dev-dependencies]
-near-sdk = { version = "5.1.0", features = ["unit-testing"] }
-near-workspaces = { version = "0.10.0", features = ["unstable"] }
+near-sdk = { version = "5.4.0", features = ["unit-testing"] }
+near-workspaces = { version = "0.11.0", features = ["unstable"] }
 tokio = { version = "1.12.0", features = ["full"] }
 serde_json = "1"
 
diff --git a/contract-rs/tests/test_basics.rs b/contract-rs/tests/test_basics.rs
index 46b5274..2e3a3ab 100644
--- a/contract-rs/tests/test_basics.rs
+++ b/contract-rs/tests/test_basics.rs
@@ -1,4 +1,4 @@
-use near_sdk::NearToken;
+use near_workspaces::types::NearToken;
 use serde_json::json;
 
 const FIVE_NEAR: NearToken = NearToken::from_near(5);