Minor tweak to tests to make the right ones pass.