diff --git a/ferritin-amplify/src/lib.rs b/ferritin-amplify/src/lib.rs
index 3c9363e..7d33d90 100644
--- a/ferritin-amplify/src/lib.rs
+++ b/ferritin-amplify/src/lib.rs
@@ -14,5 +14,6 @@
 //!
 //!
 mod amplify;
-pub use amplify::amplify::{ModelOutput, AMPLIFY};
+pub use amplify::amplify::AMPLIFY;
 pub use amplify::config::AMPLIFYConfig;
+pub use amplify::outputs::ModelOutput;
diff --git a/ferritin-core/Cargo.toml b/ferritin-core/Cargo.toml
index 0f634e0..6f74fb4 100644
--- a/ferritin-core/Cargo.toml
+++ b/ferritin-core/Cargo.toml
@@ -7,7 +7,7 @@ license.workspace = true
 description.workspace = true
 
 [dependencies]
-ferritin-pymol = { path = "../ferritin-pymol" }
+ferritin-pymol = { path = "../ferritin-pymol", optional = true }
 pdbtbx.workspace = true
 itertools.workspace = true
 lazy_static = "1.5.0"