fix test
fix test
add error check