Fix forget typehint