Fix another type error