diff --git a/qt-models/cylindermodel.h b/qt-models/cylindermodel.h
index 099a3beb2..b8ed7e0eb 100644
--- a/qt-models/cylindermodel.h
+++ b/qt-models/cylindermodel.h
@@ -5,7 +5,7 @@
 #include <QSortFilterProxyModel>
 
 #include "cleanertablemodel.h"
-#include "core/dive.h"
+#include "core/equipment.h"
 
 /* Encapsulation of the Cylinder Model, that presents the
  * Current cylinders that are used on a dive. */
diff --git a/qt-models/weightmodel.h b/qt-models/weightmodel.h
index b1df4fc8e..6d0b4e0ac 100644
--- a/qt-models/weightmodel.h
+++ b/qt-models/weightmodel.h
@@ -3,7 +3,7 @@
 #define WEIGHTMODEL_H
 
 #include "cleanertablemodel.h"
-#include "core/dive.h"
+#include "core/equipment.h"
 
 /* Encapsulation of the Weight Model, that represents
  * the current weights on a dive. */