Fix bugs

Fix bugs