improve test
fix test format