Fix #10

うまく動かなかったので、手を加えてみました。
見てみてください。