Logic fix. Add comment.