diff --git a/src/MainWindow.cpp b/src/MainWindow.cpp
index 89f902f..4b2b366 100644
--- a/src/MainWindow.cpp
+++ b/src/MainWindow.cpp
@@ -155,6 +155,7 @@ void MainWindow::createSketch()
 
 void MainWindow::saveSketch()
 {
+    m_viewport->saveSketch();
     m_tabWidget->removeTab(m_tabWidget->indexOf(m_sketchTab));
     m_tabWidget->addTab(m_solidTab, "SOLID");
     m_tabWidget->addTab(m_surfaceTab, "SURFACE");
diff --git a/src/ViewportWidget.cpp b/src/ViewportWidget.cpp
index e8e2e30..7c36de2 100644
--- a/src/ViewportWidget.cpp
+++ b/src/ViewportWidget.cpp
@@ -147,6 +147,12 @@ void ViewportWidget::startSketch(SketchPlane plane)
     update();
 }
 
+void ViewportWidget::saveSketch()
+{
+    m_currentPlane = SketchPlane::NONE;
+    update();
+}
+
 QVector3D ViewportWidget::project(const QVector3D& worldCoord, const QMatrix4x4& modelView, const QMatrix4x4& projection, const QRect& viewport)
 {
     QVector4D clipCoord = projection * modelView * QVector4D(worldCoord, 1.0);
diff --git a/src/ViewportWidget.h b/src/ViewportWidget.h
index 4e47d50..e8493ce 100644
--- a/src/ViewportWidget.h
+++ b/src/ViewportWidget.h
@@ -29,6 +29,7 @@ public:
     ~ViewportWidget();
 
     void startSketch(SketchPlane plane);
+    void saveSketch();
     void setDocument(Document* document);
 
 protected: