Fix type mismatch. Oops.