Add missing type hint