fix bugs, fix test bugs