[feature] fix some internal API