minor bug clean up