Fixing a mistake in the tests.