diff --git a/editor/src/document/document_file.rs b/editor/src/document/document_message_handler.rs
similarity index 100%
rename from editor/src/document/document_file.rs
rename to editor/src/document/document_message_handler.rs
diff --git a/editor/src/document/mod.rs b/editor/src/document/mod.rs
index 6dd71247..1cc500d5 100644
--- a/editor/src/document/mod.rs
+++ b/editor/src/document/mod.rs
@@ -1,6 +1,6 @@
 mod artboard_message_handler;
-mod document_file;
-pub mod layer_panel;
+mod document_message_handler;
+mod layer_panel;
 mod movement_handler;
 mod overlay_message_handler;
 mod portfolio_message_handler;
@@ -8,18 +8,22 @@ mod transform_layer_handler;
 mod vectorize_layer_metadata;
 
 #[doc(inline)]
-pub use document_file::LayerMetadata;
+pub use document_message_handler::{AlignAggregate, AlignAxis, DocumentMessage, DocumentMessageDiscriminant, DocumentMessageHandler, FlipAxis, VectorManipulatorSegment, VectorManipulatorShape};
 
 #[doc(inline)]
-pub use document_file::{AlignAggregate, AlignAxis, DocumentMessage, DocumentMessageDiscriminant, DocumentMessageHandler, FlipAxis, VectorManipulatorSegment, VectorManipulatorShape};
+pub use layer_panel::{LayerDataTypeDiscriminant, LayerMetadata, LayerPanelEntry, RawBuffer};
+
 #[doc(inline)]
 pub use movement_handler::{MovementMessage, MovementMessageDiscriminant};
+
 #[doc(inline)]
 pub use overlay_message_handler::{OverlayMessage, OverlayMessageDiscriminant};
+
 #[doc(inline)]
 pub use portfolio_message_handler::{Clipboard, PortfolioMessage, PortfolioMessageDiscriminant, PortfolioMessageHandler};
 
 #[doc(inline)]
 pub use artboard_message_handler::{ArtboardMessage, ArtboardMessageDiscriminant};
+
 #[doc(inline)]
 pub use transform_layer_handler::{TransformLayerMessage, TransformLayerMessageDiscriminant};
diff --git a/editor/src/frontend/frontend_message_handler.rs b/editor/src/frontend/frontend_message_handler.rs
index 5eb6b9cc..412715c1 100644
--- a/editor/src/frontend/frontend_message_handler.rs
+++ b/editor/src/frontend/frontend_message_handler.rs
@@ -1,4 +1,4 @@
-use crate::document::layer_panel::{LayerPanelEntry, RawBuffer};
+use crate::document::{LayerPanelEntry, RawBuffer};
 use crate::message_prelude::*;
 use crate::misc::HintData;
 use crate::tool::tool_options::ToolOptions;