fix a TODO(if) statement