1 files changed, 7 insertions, 2 deletions
diff --git a/qt-ui/mainwindow.cpp b/qt-ui/mainwindow.cpp
index aef22497b..e4309a23b 100644
--- a/qt-ui/mainwindow.cpp
+++ b/qt-ui/mainwindow.cpp
@@ -215,8 +215,13 @@ void MainWindow::on_actionManage_dive_sites_triggered() {
 	enableDiveSiteEdit(displayed_dive.dive_site_uuid);
 }
 
-void MainWindow::enableDiveSiteEdit(uint32_t id) {
-	locationInformationWidget()->setCurrentDiveSiteByUuid(id);
+void MainWindow::enableDiveSiteCreation() {
+	locationInformationWidget()->createDiveSite();
+	setApplicationState("EditDiveSite");
+}
+
+ void MainWindow::enableDiveSiteEdit(uint32_t id) {
+	locationInformationWidget()->editDiveSite(id);
 	setApplicationState("EditDiveSite");
 }