Fix tests with assertTag

assertTag is deprecated and throws an deprecated message, failing the tests.
Replaced it with a simple Contains assertion.