Add a missing type hint