diff --git a/program/tests/deploy.rs b/program/tests/deploy.rs
index 33e65dc..c9862aa 100644
--- a/program/tests/deploy.rs
+++ b/program/tests/deploy.rs
@@ -670,6 +670,7 @@ fn success() {
         ],
         &[
             Check::success(),
+            Check::compute_units(19_893),
             Check::account(&program)
                 .data(
                     &bincode::serialize(&UpgradeableLoaderState::Program {
diff --git a/program/tests/initialize_buffer.rs b/program/tests/initialize_buffer.rs
index e283acc..c96b513 100644
--- a/program/tests/initialize_buffer.rs
+++ b/program/tests/initialize_buffer.rs
@@ -110,6 +110,7 @@ fn success() {
         ],
         &[
             Check::success(),
+            Check::compute_units(2_168),
             Check::account(&source)
                 .lamports(100_000_000)
                 .owner(&solana_loader_v3_program::id())
diff --git a/program/tests/write.rs b/program/tests/write.rs
index c59588c..5644200 100644
--- a/program/tests/write.rs
+++ b/program/tests/write.rs
@@ -209,6 +209,7 @@ fn success() {
         ],
         &[
             Check::success(),
+            Check::compute_units(1_419),
             Check::account(&buffer).data(&check_data).build(),
         ],
     );
@@ -227,6 +228,7 @@ fn success() {
         ],
         &[
             Check::success(),
+            Check::compute_units(1_391),
             Check::account(&buffer).data(&check_data).build(),
         ],
     );