Fix bug for single line FOR