diff --git a/Dockerfile b/Dockerfile
index c6c02e4..6b56c79 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 ARG BUILD_PLATFORM=linux/amd64
-ARG RUST_VERSION=1.83
+ARG RUST_VERSION=1.81
 FROM --platform=${BUILD_PLATFORM} rust:${RUST_VERSION} AS builder
 
 RUN rustup target add x86_64-unknown-linux-gnu
diff --git a/main/src/main.rs b/main/src/main.rs
index d4c0c52..b8bf51e 100644
--- a/main/src/main.rs
+++ b/main/src/main.rs
@@ -1,6 +1,9 @@
 // Copyright 2023-2024, Offchain Labs, Inc.
 // For licensing, see https://github.com/OffchainLabs/cargo-stylus/blob/main/licenses/COPYRIGHT.md
 
+// Enable unstable test feature for benchmarks when nightly is available
+#![cfg_attr(feature = "nightly", feature(test))]
+
 use alloy_primitives::TxHash;
 use clap::{ArgGroup, Args, CommandFactory, Parser, Subcommand};
 use constants::DEFAULT_ENDPOINT;
diff --git a/rust-toolchain.toml b/rust-toolchain.toml
index 0193dee..1de01fa 100644
--- a/rust-toolchain.toml
+++ b/rust-toolchain.toml
@@ -1,2 +1,2 @@
 [toolchain]
-channel = "1.83.0"
+channel = "1.81.0"