fix bugs # 3