added patch for task stage initialization