diff --git a/near-sdk/tests/store_performance_tests.rs b/near-sdk/tests/store_performance_tests.rs
index 222f27e4e..f3458c228 100644
--- a/near-sdk/tests/store_performance_tests.rs
+++ b/near-sdk/tests/store_performance_tests.rs
@@ -459,7 +459,7 @@ async fn test_lazy() -> anyhow::Result<()> {
 
     let res = account
         .call(&contract_id, "flush")
-        .args_json((3550000,))
+        .args_json((3000000,))
         .max_gas()
         .transact()
         .await?
@@ -488,13 +488,13 @@ async fn test_lazy() -> anyhow::Result<()> {
     perform_asserts(res.total_gas_burnt.as_gas(), "lazy:insert_flush");
 
     let res = account
-        .call(&contract_id, "insert_take_flush")
+        .call(&contract_id, "insert_take")
         .args_json((700,))
         .max_gas()
         .transact()
         .await?
         .unwrap();
 
-    perform_asserts(res.total_gas_burnt.as_gas(), "lazy:insert_take_flush");
+    perform_asserts(res.total_gas_burnt.as_gas(), "lazy:insert_take");
     Ok(())
 }
diff --git a/near-sdk/tests/test-contracts/lazy/src/lib.rs b/near-sdk/tests/test-contracts/lazy/src/lib.rs
index 8f204d960..6d2ac49a7 100644
--- a/near-sdk/tests/test-contracts/lazy/src/lib.rs
+++ b/near-sdk/tests/test-contracts/lazy/src/lib.rs
@@ -60,7 +60,7 @@ impl LazyContract {
 
     /// This should write twice on each iteration.
     #[payable]
-    pub fn insert_take_flush(&mut self, iterations: u32) {
+    pub fn insert_take(&mut self, iterations: u32) {
         let mut insertable = self.insertable();
         for idx in 0..=iterations {
             insertable.index = idx as u32;