Fix unit test. See #344.