Minor bug corrected