fix bugs +