fixup: make test more robust