Fixed the loop control