fix for wrong return type hint