Bug Fix in work_space.py