fix some typehinting