fix bugs - 2