diff --git a/dist/flatland_rl-3.0.7-py2.py3-none-any.whl b/dist/flatland_rl-3.0.8-py2.py3-none-any.whl
similarity index 98%
rename from dist/flatland_rl-3.0.7-py2.py3-none-any.whl
rename to dist/flatland_rl-3.0.8-py2.py3-none-any.whl
index e851deb6af225b20cbdb300abedfb3e78abf73f7..4798fa05d119ce7d78ca8d44200b5d0518e73a9f 100644
Binary files a/dist/flatland_rl-3.0.7-py2.py3-none-any.whl and b/dist/flatland_rl-3.0.8-py2.py3-none-any.whl differ
diff --git a/flatland/__init__.py b/flatland/__init__.py
index ac366bdf8bb1091755daca91a411a90aa36018f2..e7369188ee2a50f959e7ae4bd97bbed9145b5d83 100644
--- a/flatland/__init__.py
+++ b/flatland/__init__.py
@@ -4,4 +4,4 @@
 
 __author__ = """S.P. Mohanty"""
 __email__ = 'mohanty@aicrowd.com'
-__version__ = '3.0.7'
+__version__ = '3.0.8'
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 91fab9cacd46a8cf00085b4a0f8e30aa0fa4e4fd..91726cddb6bf9cc166849f79ec140940f6991b15 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -494,6 +494,7 @@ class RailEnv(Environment):
         """ Any updates to agent to be made in Done state """
         if agent.state == TrainState.DONE and agent.arrival_time is None:
             agent.arrival_time = self._elapsed_steps
+            self.dones[agent.handle] = True
             if self.remove_agents_at_target:
                 agent.position = None
 
diff --git a/setup.cfg b/setup.cfg
index a0f6a0f5a185926f66916d9a6f6bcb95536d010a..fda27bdfa5861c8522f1be1930f7333780e7ba37 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 3.0.7
+current_version = 3.0.8
 commit = True
 tag = True
 
diff --git a/setup.py b/setup.py
index cea53d34860a43ef4565e6e432449f55c73ceeee..f75051aff9867a48123df9b590b8820fd8338754 100644
--- a/setup.py
+++ b/setup.py
@@ -80,6 +80,6 @@ setup(
     test_suite='tests',
     tests_require=test_requirements,
     url='https://gitlab.aicrowd.com/flatland/flatland',
-    version='3.0.7',
+    version='3.0.8',
     zip_safe=False,
 )