Use assertTag

This fixes one of the tests