From 352ba23f83adb5bddb5c65d1d2e6637a46cda05d Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Sun, 14 Mar 2021 20:49:56 -0700
Subject: [PATCH] revert previous change

revert 29e62aba00291351076d9489f4444af87b5e1d81
---
 weed/shell/command_volume_balance.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/weed/shell/command_volume_balance.go b/weed/shell/command_volume_balance.go
index 33ae9344d..e0c41f310 100644
--- a/weed/shell/command_volume_balance.go
+++ b/weed/shell/command_volume_balance.go
@@ -201,7 +201,7 @@ type CapacityFunc func(*master_pb.DataNodeInfo) int
 
 func capacityByMaxVolumeCount(diskType types.DiskType) CapacityFunc {
 	return func(info *master_pb.DataNodeInfo) int {
-		diskInfo, found := info.DiskInfos[diskType.ReadableString()]
+		diskInfo, found := info.DiskInfos[string(diskType)]
 		if !found {
 			return 0
 		}