fix two small bugs~