test: tweak test
refactor to use "within-element"

Simplifies the code to just 1 line :p