Fix wrong typehint