Minor bug correction