fix bugs .....