fix bugs
Fix bug