Fix incorrect type hint