Fix test.
This isn't pretty.