diff --git a/docs/storage/driver/testsuites/testsuites.go b/docs/storage/driver/testsuites/testsuites.go
index 48d90ed8f..de8e31432 100644
--- a/docs/storage/driver/testsuites/testsuites.go
+++ b/docs/storage/driver/testsuites/testsuites.go
@@ -412,12 +412,12 @@ func (suite *DriverSuite) testContinueStreamAppend(c *check.C, chunkSize int64)
 	c.Assert(err, check.IsNil)
 	c.Assert(nn, check.Equals, int64(len(contentsChunk1)))
 
-	curSize := writer.Size()
-	c.Assert(curSize, check.Equals, int64(len(contentsChunk1)))
-
 	err = writer.Close()
 	c.Assert(err, check.IsNil)
 
+	curSize := writer.Size()
+	c.Assert(curSize, check.Equals, int64(len(contentsChunk1)))
+
 	writer, err = suite.StorageDriver.Writer(suite.ctx, filename, true)
 	c.Assert(err, check.IsNil)
 	c.Assert(writer.Size(), check.Equals, curSize)
@@ -426,12 +426,12 @@ func (suite *DriverSuite) testContinueStreamAppend(c *check.C, chunkSize int64)
 	c.Assert(err, check.IsNil)
 	c.Assert(nn, check.Equals, int64(len(contentsChunk2)))
 
-	curSize = writer.Size()
-	c.Assert(curSize, check.Equals, 2*chunkSize)
-
 	err = writer.Close()
 	c.Assert(err, check.IsNil)
 
+	curSize = writer.Size()
+	c.Assert(curSize, check.Equals, 2*chunkSize)
+
 	writer, err = suite.StorageDriver.Writer(suite.ctx, filename, true)
 	c.Assert(err, check.IsNil)
 	c.Assert(writer.Size(), check.Equals, curSize)