Update

Fix bug