Slight improvement to unit test.