diff --git a/common/types/mapper/proto/sharddistributor.go b/common/types/mapper/proto/sharddistributor.go
index b5ac4f27baa..a95309d8171 100644
--- a/common/types/mapper/proto/sharddistributor.go
+++ b/common/types/mapper/proto/sharddistributor.go
@@ -104,6 +104,8 @@ func FromShardDistributorExecutorHeartbeatRequest(t *types.ExecutorHeartbeatRequ
 				status = sharddistributorv1.ShardStatus_SHARD_STATUS_INVALID
 			case types.ShardStatusREADY:
 				status = sharddistributorv1.ShardStatus_SHARD_STATUS_READY
+			case types.ShardStatusDONE:
+				status = sharddistributorv1.ShardStatus_SHARD_STATUS_DONE
 			default:
 				status = sharddistributorv1.ShardStatus_SHARD_STATUS_INVALID
 			}