Fix type hints in the relations tests.