Fix tests for task module