fix mistake in the tests