fix bugs