Add missing type hinting