Typehint fix