Fix inifinite loop