diff --git a/crates/format/src/input.rs b/crates/format/src/input.rs
index c0f70a5..cc96341 100644
--- a/crates/format/src/input.rs
+++ b/crates/format/src/input.rs
@@ -25,9 +25,9 @@ use crate::{metadata::ContractInstance, traits::ResolutionContext};
#[serde(untagged)]
pub enum Step {
/// A function call or an invocation to some function on some smart contract.
- FunctionCall(Input),
+ FunctionCall(Box),
/// A step for performing a balance assertion on some account or contract.
- BalanceAssertion(BalanceAssertion),
+ BalanceAssertion(Box),
}
#[derive(Clone, Debug, Default, Serialize, Deserialize, Eq, PartialEq)]