fix single ddp case