Fix bugs ..