1 files changed, 3 insertions, 3 deletions
diff --git a/qt-ui/maintab.h b/qt-ui/maintab.h
index 8c398a77d..5180b26e2 100644
--- a/qt-ui/maintab.h
+++ b/qt-ui/maintab.h
@@ -9,11 +9,12 @@
 
 #include <QTabWidget>
 #include <QDialog>
+#include <QMap>
 
 #include "models.h"
 
 class QCompleter;
-
+struct dive;
 namespace Ui
 {
 	class MainTab;
@@ -76,8 +77,7 @@ private:
 	Ui::MainTab *ui;
 	WeightModel *weightModel;
 	CylindersModel *cylindersModel;
-	NotesBackup notesBackup;
-	struct dive* currentDive;
+	QMap<dive*, NotesBackup> notesBackup;
 	QPushButton *addCylinder;
 	QPushButton *addWeight;
 	enum { NONE, DIVE, TRIP } editMode;