diff --git a/microsoft-azure-api/src/test/java/com/microsoft/windowsazure/services/blob/client/CloudBlobContainerTests.java b/microsoft-azure-api/src/test/java/com/microsoft/windowsazure/services/blob/client/CloudBlobContainerTests.java
index 9d945f61a53a9..89c983c3852ad 100644
--- a/microsoft-azure-api/src/test/java/com/microsoft/windowsazure/services/blob/client/CloudBlobContainerTests.java
+++ b/microsoft-azure-api/src/test/java/com/microsoft/windowsazure/services/blob/client/CloudBlobContainerTests.java
@@ -748,21 +748,21 @@ public void testCurrentOperationByteCount() throws URISyntaxException, StorageEx
 
         OperationContext operationContext = new OperationContext();
         BlobRequestOptions options = new BlobRequestOptions();
-        options.setTimeoutIntervalInMs(1000);
+        options.setTimeoutIntervalInMs(2000);
         options.setRetryPolicyFactory(new RetryNoRetry());
+        ByteArrayOutputStream downloadedDataStream = new ByteArrayOutputStream();
         try {
-            final ByteArrayOutputStream downloadedDataStream = new ByteArrayOutputStream();
             blobRef.download(downloadedDataStream, null, options, operationContext);
         }
         catch (Exception e) {
-            Assert.assertEquals(0, operationContext.getCurrentOperationByteCount());
+            Assert.assertEquals(downloadedDataStream.size(), operationContext.getCurrentOperationByteCount());
         }
 
         operationContext = new OperationContext();
         options = new BlobRequestOptions();
         options.setTimeoutIntervalInMs(90000);
 
-        final ByteArrayOutputStream downloadedDataStream = new ByteArrayOutputStream();
+        downloadedDataStream = new ByteArrayOutputStream();
         blobRef.download(downloadedDataStream, null, options, operationContext);
 
         Assert.assertEquals(blockLength * numberOfBlocks, operationContext.getCurrentOperationByteCount());