[feature] fix some code