Fix mistake in tests