fix bugs ...