diff --git a/src/main/java/de/uniluebeck/mi/projmi6/db/DBHandler.java b/src/main/java/de/uniluebeck/mi/projmi6/db/DBHandler.java
index 16ca386..abf6b4f 100644
--- a/src/main/java/de/uniluebeck/mi/projmi6/db/DBHandler.java
+++ b/src/main/java/de/uniluebeck/mi/projmi6/db/DBHandler.java
@@ -485,8 +485,8 @@ public class DBHandler {
             statement.setString(5, null);
         }
         statement.setBoolean(6, fall.getSelbsteinweisung());                            // Selbsteinweisung
-        if (fall.getHauptDiagnose() != null) {
-            statement.setInt(7, fall.getHauptDiagnose().getDiagID());                   // Hauptdiagnose
+        if (fall.getHauptdiagnoseId() != 0) {
+            statement.setInt(7, fall.getHauptdiagnoseId());                   // Hauptdiagnose
         } else {
             statement.setTimestamp(7, null);
         }
@@ -546,6 +546,7 @@ public class DBHandler {
     private static Diagnose getDiagnose(ResultSet rs) throws SQLException {
         Diagnose diagnose = new Diagnose();
 
+        diagnose.setDiagID(rs.getInt("diagid"));
         diagnose.setFreiText(rs.getString("freitext"));
         diagnose.setArzt(new Mitarbeiter(rs.getInt("arzt")));
         diagnose.setDiagArt(DiagArt.parseInt(rs.getInt("diagart")));