fix package
fix test