diff --git a/pkg/controller/framework_test.go b/pkg/controller/framework_test.go
index f811854f6..a9e8450b5 100644
--- a/pkg/controller/framework_test.go
+++ b/pkg/controller/framework_test.go
@@ -752,7 +752,7 @@ func newContent(name, className, snapshotHandle, volumeUID, volumeName, boundToS
 		Spec: crdv1.VolumeSnapshotContentSpec{
 			VolumeSnapshotSource: crdv1.VolumeSnapshotSource{
 				CSI: &crdv1.CSIVolumeSnapshotSource{
-					Size:           size,
+					RestoreSize:    size,
 					Driver:         mockDriverName,
 					SnapshotHandle: snapshotHandle,
 					CreationTime:   creationTime,
@@ -807,7 +807,7 @@ func newSnapshot(name, className, boundToContent, snapshotUID, claimName string,
 			CreationTime: creationTime,
 			Ready:        ready,
 			Error:        err,
-			Size:         size,
+			RestoreSize:  size,
 		},
 	}