Afghanistan forces prepare to retake Taliban-held district