Minor bug correction

Bug correction.