Minor tweaks in test.