Minor bug correction.