fixed if loop error