Fix bad typing