Update test and fix test error.