diff --git a/editor/src/document/document_message_handler.rs b/editor/src/document/document_message_handler.rs
index c419ed96..1d5598b1 100644
--- a/editor/src/document/document_message_handler.rs
+++ b/editor/src/document/document_message_handler.rs
@@ -1182,7 +1182,7 @@ impl MessageHandler<DocumentMessage, &InputPreprocessorMessageHandler> for Docum
 				responses.push_back(ToolMessage::DocumentIsDirty.into());
 			}
 			Undo => {
-				responses.push_back(SelectToolMessage::Abort.into());
+				responses.push_back(ToolMessage::AbortCurrentTool.into());
 				responses.push_back(DocumentHistoryBackward.into());
 				responses.push_back(ToolMessage::DocumentIsDirty.into());
 				responses.push_back(RenderDocument.into());