improve test check
improve test check