Trying to make test more robust.

Not really sure what's wrong here.