fix wrong type hint