Fixed the IF loop?