Minor adjustments to test.