diff --git a/DG_Approximation.py b/DG_Approximation.py
index d89a79bdeab8f2e59013815aeb21f9813d0e6a48..7493eff3bfa98e38a71ba5c268eaa9525c2d3b49 100644
--- a/DG_Approximation.py
+++ b/DG_Approximation.py
@@ -27,7 +27,7 @@ TODO: Check whether consistency is given/possible for each class instance
 
 TODO: Make projection local variable -> Done
 TODO: Vector faster than Trans for longer processes, therefore replace -> Done
-TODO: Make sure time is changed to current_time everywhere 
+TODO: Make sure time is changed to current_time everywhere -> Done
 
 """
 import numpy as np
diff --git a/Update_Scheme.py b/Update_Scheme.py
index fb63815bca28bfefbe0b93ae1b0227439e20b1ad..e0f56544fd8bc144713f7c3aa7f7a33a333f6efc 100644
--- a/Update_Scheme.py
+++ b/Update_Scheme.py
@@ -43,11 +43,11 @@ class UpdateScheme(object):
     def get_time_history(self):
         return self.time_history
 
-    def step(self, projection, cfl_number, time):
+    def step(self, projection, cfl_number, current_time):
         self.original_projection = projection
         self.current_projection = projection
         self.cfl_number = cfl_number
-        self.time = time
+        self.time = current_time
 
         self._apply_stability_method()