Fix broken code
Merge pull request #1 from yuya-takeyama/patch-1

Fix broken code