Complete typehint fix