diff --git a/crates/sdk/src/keygen/dummy.rs b/crates/sdk/src/keygen/dummy.rs
index 4a9b20e4db..07a436c80e 100644
--- a/crates/sdk/src/keygen/dummy.rs
+++ b/crates/sdk/src/keygen/dummy.rs
@@ -54,7 +54,7 @@ use crate::{
 /// - trace heights ordered by AIR ID
 ///
 /// All trace heights are rounded to the next power of two (or 0 -> 0).
-pub(super) fn compute_root_proof_heights(
+pub fn compute_root_proof_heights(
     root_vm: &mut VirtualMachine<BabyBearPoseidon2RootEngine, NativeCpuBuilder>,
     root_committed_exe: &VmCommittedExe<BabyBearPoseidon2RootConfig>,
     dummy_internal_proof: &Proof<SC>,
diff --git a/crates/sdk/src/keygen/mod.rs b/crates/sdk/src/keygen/mod.rs
index ddf0c2b570..89d049a34c 100644
--- a/crates/sdk/src/keygen/mod.rs
+++ b/crates/sdk/src/keygen/mod.rs
@@ -54,7 +54,7 @@ use crate::{
 };
 
 pub mod asm;
-pub(crate) mod dummy;
+pub mod dummy;
 pub mod perm;
 #[cfg(feature = "evm-prove")]
 pub mod static_verifier;
diff --git a/crates/sdk/src/keygen/perm.rs b/crates/sdk/src/keygen/perm.rs
index f52d1ebd12..c4c7012d97 100644
--- a/crates/sdk/src/keygen/perm.rs
+++ b/crates/sdk/src/keygen/perm.rs
@@ -4,7 +4,7 @@ use std::cmp::Reverse;
 use openvm_continuations::verifier::common::types::SpecialAirIds;
 
 /// Permutation of the AIR IDs to order them by forced trace heights.
-pub(crate) struct AirIdPermutation {
+pub struct AirIdPermutation {
     pub perm: Vec<usize>,
 }