fix two small bugs
Changes