fixed return type hint