fixed mistake in tests