Fix bad type hintings