improve unittest