1 files changed, 4 insertions, 2 deletions
diff --git a/qt-ui/models.h b/qt-ui/models.h
index 41bd4f892..62ae3913c 100644
--- a/qt-ui/models.h
+++ b/qt-ui/models.h
@@ -70,9 +70,11 @@ public:
 	void add(weightsystem_t *weight);
 	void clear();
 	void update();
+	void setDive(struct dive *d);
+
 private:
-	/* Remember the number of rows in a dive */
-	QMap<struct dive *, int> usedRows;
+	struct dive *current;
+	int rows;
 };
 
 /*! An AbstractItemModel for recording dive trip information such as a list of dives.