Fixed mistake in test