diff --git a/contract/pkg/messages.go b/contract/pkg/messages.go
index f253abb..d5ecc16 100644
--- a/contract/pkg/messages.go
+++ b/contract/pkg/messages.go
@@ -35,15 +35,9 @@ type NodeStatus struct {
 type Bucket struct {
 	OwnerId          types.AccountID
 	ClusterId        uint32
-	Flow             Flow
 	ResourceReserved uint32
 }
 
-type Flow struct {
-	From     types.AccountID
-	Schedule Schedule
-}
-
 type Schedule struct {
 	Rate   types.U128
 	Offset types.U128
diff --git a/contract/pkg/mock/ddc_bucket_contract_mock.go b/contract/pkg/mock/ddc_bucket_contract_mock.go
index 87dc89c..42f0a53 100644
--- a/contract/pkg/mock/ddc_bucket_contract_mock.go
+++ b/contract/pkg/mock/ddc_bucket_contract_mock.go
@@ -113,7 +113,6 @@ func CreateBucket(bucketId uint32, bucketParams string) *pkg.BucketStatus {
 		Bucket: pkg.Bucket{
 			OwnerId:          types.AccountID{},
 			ClusterId:        0,
-			Flow:             pkg.Flow{},
 			ResourceReserved: 0,
 		},
 		Params:             bucketParams,