fix som bugs
Fix