fix test bugs