diff --git a/CADETProcess/processModel/flowSheet.py b/CADETProcess/processModel/flowSheet.py
index 706f65e0..29fdd4ea 100644
--- a/CADETProcess/processModel/flowSheet.py
+++ b/CADETProcess/processModel/flowSheet.py
@@ -517,11 +517,11 @@ def set_output_state(self, unit, state):
             if state >= state_length:
                 raise CADETProcessError('Index exceeds destinations')
 
-            output_state = [0] * state_length
-            output_state[state] = 1
+            output_state = [0.0] * state_length
+            output_state[state] = 1.0
 
         elif isinstance(state, dict):
-            output_state = [0] * state_length
+            output_state = [0.0] * state_length
             for dest, value in state.items():
                 try:
                     assert self.connection_exists(unit, dest)