make test more flexible