fixing type hint