made test a little more flexible (code already supports this)