Corrected a small flaw