Fix type hint of returns value