Corrected another bug