Fix bug
Bug fixes