fix bugs...