Fix type mismatch warning in previous commit