CS版本的修复

Signed-off-by: MagicHu <huweimin@cn.ibm.com>