Fixed the bug

Close #1