bug fixds