Minor bug fix
Merge pull request #1 from ryush00/v0.1

Minor bug fix