Fix code
Merge pull request #1637 from Hikkijp/patch-3

Fix code