Fixed if loop error