Fix bugs .....