#47 Minor bug correction