Fix argument type hint