fix: type hint should use Union

fix: type hint should use Union #627