Fixing bug in tests

That's what you get from copying and pasting and forgetting to
modify the pasted lines to suit your needs.