Fixed silly bug in tests.