[feature] fix tests