fix type hint