Fix another type checking error