You should just be able to specify a class to match in the block, so we probably don't need this