fix for the task wrapper