minor bug correction