get the test more strict
refactor test