Small bug corrected