Fix Bugs

Fix Bugs