Fix type in comments