Fix test.

Fix suggested by Edd Barrett (and I'm an idiot for not thinking of it!).