Correct mistake in tests