Bug correction

Minor bug correction