fixed silly mistake in tests