Fix bugs...