fix tests cases
fix tests cases