Minor correction with tests.