fixed issue in work_controller