diff --git a/LDDsimulation/LDDsimulation.py b/LDDsimulation/LDDsimulation.py
index 33cb47cd049fe1131459dcbeb30fc2be0fdfb10c..0d9b8fa87b03eda44733b4abb3af563d40120f63 100644
--- a/LDDsimulation/LDDsimulation.py
+++ b/LDDsimulation/LDDsimulation.py
@@ -1045,7 +1045,7 @@ class LDDsimulation(object):
                         subdomain.function_space["pressure"]['wetting']
                         )
                     saturation_nw = df.Function(
-                        subdomain.function_space["pressure"]['wetting']
+                        subdomain.function_space["pressure"]['nonwetting']
                         )
 
                 for phase in subdomain.has_phases:
@@ -1062,7 +1062,6 @@ class LDDsimulation(object):
                         "source_"+"{}".format(phase)
                         )
                     file.write(source[phase], timestep)
-
                     if self.exact_solution:
                         pa_exact = subdomain.pressure_exact[phase]
                         pa_exact.t = timestep