fix bugs ..