fix bugs

cuz #4