Small Bug correction