Fix bug
Fix bug 1