fix simple batch sample