making test more flexible