diff --git a/crates/chia-consensus/Cargo.toml b/crates/chia-consensus/Cargo.toml
index a57427d4c..7b740a0fa 100644
--- a/crates/chia-consensus/Cargo.toml
+++ b/crates/chia-consensus/Cargo.toml
@@ -13,6 +13,8 @@ workspace = true
 
 [features]
 py-bindings = ["dep:pyo3", "dep:chia_py_streamable_macro"]
+chiavdf = ["dep:chiavdf"]
+chiapos = ["dep:chiapos"]
 
 [dependencies]
 clvmr = { workspace = true }
@@ -29,8 +31,8 @@ chia-puzzles = { workspace = true }
 chia-bls = { workspace = true }
 hex-literal = { workspace = true }
 thiserror = { workspace = true }
-chiavdf = { workspace = true }
-chiapos = { workspace = true }
+chiavdf = { workspace = true, optional = true }
+chiapos = { workspace = true, optional = true }
 
 [dev-dependencies]
 num-traits = { workspace = true }
diff --git a/crates/chia-consensus/src/lib.rs b/crates/chia-consensus/src/lib.rs
index f8d391b05..c35df1413 100644
--- a/crates/chia-consensus/src/lib.rs
+++ b/crates/chia-consensus/src/lib.rs
@@ -8,7 +8,11 @@ pub mod gen;
 pub mod generator_rom;
 pub mod merkle_set;
 pub mod merkle_tree;
-pub mod proof_of_space;
 pub mod spendbundle_conditions;
 pub mod spendbundle_validation;
+
+#[cfg(feature = "chiavdf")]
 pub mod vdf;
+
+#[cfg(feature = "chiapos")]
+pub mod proof_of_space;
diff --git a/wheel/Cargo.toml b/wheel/Cargo.toml
index f3d602ddd..298d55573 100644
--- a/wheel/Cargo.toml
+++ b/wheel/Cargo.toml
@@ -23,9 +23,8 @@ python-source = "python"
 [dependencies]
 clvmr = { workspace = true }
 hex = { workspace = true }
-
 pyo3 = { workspace = true, features = ["multiple-pymethods"] }
-chia-consensus = { workspace = true, features = ["py-bindings"] }
+chia-consensus = { workspace = true, features = ["py-bindings", "chiavdf", "chiapos"] }
 chia-bls = { workspace = true, features = ["py-bindings"]  }
 chia-protocol = { workspace = true, features = ["py-bindings"]  }
 chia-traits = { workspace = true, features = ["py-bindings"]  }