diff --git a/libs/jwst-storage/src/entities/prelude.rs b/libs/jwst-storage/src/entities/prelude.rs
index f5578d70..af39fb91 100644
--- a/libs/jwst-storage/src/entities/prelude.rs
+++ b/libs/jwst-storage/src/entities/prelude.rs
@@ -1,6 +1,7 @@
 //! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.6
 
-pub use super::blobs::Entity as Blobs;
+#[cfg(feature = "bucket")]
 pub use super::bucket_blobs::Entity as BucketBlobs;
-pub use super::docs::Entity as Docs;
+#[cfg(feature = "image")]
 pub use super::optimized_blobs::Entity as OptimizedBlobs;
+pub use super::{blobs::Entity as Blobs, docs::Entity as Docs};