fix a couple of small bugs