Change if inside main Setup