Fixed oversight in tests.