diff --git a/src/main/java/fi/lpam/dataluokat/Kuljetus.java b/src/main/java/fi/lpam/dataluokat/Kuljetus.java
index e5efe97..5dad57f 100644
--- a/src/main/java/fi/lpam/dataluokat/Kuljetus.java
+++ b/src/main/java/fi/lpam/dataluokat/Kuljetus.java
@@ -44,9 +44,9 @@ public class Kuljetus implements TietokantaOlio {
         this.kuljetusPäivämäärä = pvm;
         this.kuljetusKierros = kuljetusKierros;
         this.tyyppi = tyyppi;
-        this.salaatit = asiakas.getPäivänSalaatit(pvm);
-        this.pääruoat = asiakas.getPäivänPääruoat(pvm);
-        this.jälkiruoat = asiakas.getPäivänJälkiruoat(pvm);
+        this.salaatit = asiakas.getKuljetuksenSalaatit(pvm, tyyppi);
+        this.pääruoat = asiakas.getKuljetuksenPääruoat(pvm, tyyppi);
+        this.jälkiruoat = asiakas.getKuljetuksenJälkiruoat(pvm, tyyppi);
         this.osoite = asiakas.getOsoite();
         this.avainTieto = asiakas.getAvainTieto();
         this.lisätieto = asiakas.getLisätiedot();
@@ -190,12 +190,6 @@ public class Kuljetus implements TietokantaOlio {
 
         try {
             while (rs.next()) {
-                Tyyppi kuljetusTyyppi = Tyyppi.LOUNAS;
-                switch (rs.getString("tyyppi")) {
-                    case "LOUNAS": kuljetusTyyppi = Tyyppi.LOUNAS; break;
-                    case "PÄIVÄLLINEN": kuljetusTyyppi = Tyyppi.PÄIVÄLLINEN; break;
-                }
-
                 Kuljetus uusi = new Kuljetus(
                         rs.getInt("id"),
                         rs.getInt("asiakasID"),
@@ -209,7 +203,7 @@ public class Kuljetus implements TietokantaOlio {
                         rs.getString("lisätieto"),
                         rs.getString("avainTieto"),
                         rs.getString("ruokarajoite"),
-                        kuljetusTyyppi
+                        Tyyppi.valueOf(rs.getString("tyyppi"))
                 );
                 kuljetukset.add(uusi);
             }
@@ -246,9 +240,9 @@ public class Kuljetus implements TietokantaOlio {
     }
 
     public void palautaMäärät(Asiakas asiakas) {
-        setSalaatit(asiakas.getPäivänSalaatit(kuljetusPäivämäärä));
-        setPääruoat(asiakas.getPäivänPääruoat(kuljetusPäivämäärä));
-        setJälkiruoat(asiakas.getPäivänJälkiruoat(kuljetusPäivämäärä));
+        setSalaatit(asiakas.getKuljetuksenSalaatit(kuljetusPäivämäärä, tyyppi));
+        setPääruoat(asiakas.getKuljetuksenPääruoat(kuljetusPäivämäärä, tyyppi));
+        setJälkiruoat(asiakas.getKuljetuksenJälkiruoat(kuljetusPäivämäärä, tyyppi));
         setLisätieto(asiakas.getLisätiedot());
     }