minor bug update